Ahmed commited on
Commit
2c385df
·
verified ·
1 Parent(s): ecc129a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def SummarizeThis(Dialogue):
16
  """
17
 
18
  inputs = tokenizer(prompt, return_tensors='pt')
19
- output = tokenizer.decode(
20
  model.generate(
21
  inputs["input_ids"],
22
  max_new_tokens=200,
@@ -24,7 +24,7 @@ def SummarizeThis(Dialogue):
24
  skip_special_tokens=True
25
  )
26
 
27
- return output
28
 
29
 
30
  # Making the gradio application
 
16
  """
17
 
18
  inputs = tokenizer(prompt, return_tensors='pt')
19
+ Summary = tokenizer.decode(
20
  model.generate(
21
  inputs["input_ids"],
22
  max_new_tokens=200,
 
24
  skip_special_tokens=True
25
  )
26
 
27
+ return Summary
28
 
29
 
30
  # Making the gradio application