leadingbridge commited on
Commit
b214254
1 Parent(s): fa55da5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -11,12 +11,10 @@ import torch
11
  model_path = "leadingbridge/summarization"
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
  model = T5ForConditionalGeneration.from_pretrained(model_path)
14
- text = "This is the initial testing text"
15
- inputs = tokenizer.encode(text, return_tensors="pt")
16
- summary_ids = model.generate(inputs)
17
- summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
18
  def summarize_text(text):
19
- # Tokenize the input text
 
20
  inputs = tokenizer.encode(text, return_tensors="pt")
21
  # Generate the summary
22
  summary_ids = model.generate(inputs)
 
11
  model_path = "leadingbridge/summarization"
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
  model = T5ForConditionalGeneration.from_pretrained(model_path)
14
+
 
 
 
15
  def summarize_text(text):
16
+ text = "This is the initial testing text"
17
+ # Tokenize the input text
18
  inputs = tokenizer.encode(text, return_tensors="pt")
19
  # Generate the summary
20
  summary_ids = model.generate(inputs)