PRAli22 commited on
Commit
f022e05
·
1 Parent(s): 77b38ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -12,12 +12,9 @@ def summarize(text):
12
  summarized_ids = model.generate(inputs['input_ids'], attention_mask=inputs['attention_mask'],
13
  max_length=150, num_beams=4, early_stopping=True)
14
 
15
- f= " ".join([tokenizer.decode(token_ids, skip_special_tokens=True)
16
  for token_ids in summarized_ids])
17
- if f == 0:
18
- return "negative"
19
- else:
20
- return "positive"
21
 
22
 
23
  css_code='body{background-image:url("https://media.istockphoto.com/id/1256252051/vector/people-using-online-translation-app.jpg?s=612x612&w=0&k=20&c=aa6ykHXnSwqKu31fFR6r6Y1bYMS5FMAU9yHqwwylA94=");}'
 
12
  summarized_ids = model.generate(inputs['input_ids'], attention_mask=inputs['attention_mask'],
13
  max_length=150, num_beams=4, early_stopping=True)
14
 
15
+ return " ".join([tokenizer.decode(token_ids, skip_special_tokens=True)
16
  for token_ids in summarized_ids])
17
+
 
 
 
18
 
19
 
20
  css_code='body{background-image:url("https://media.istockphoto.com/id/1256252051/vector/people-using-online-translation-app.jpg?s=612x612&w=0&k=20&c=aa6ykHXnSwqKu31fFR6r6Y1bYMS5FMAU9yHqwwylA94=");}'