PRAli22 commited on
Commit
2328652
·
1 Parent(s): f64712f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -12,11 +12,8 @@ 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= tokenizer.decode(summarized_ids[0], skip_special_tokens=True)
16
- if f == 1:
17
- return "positive"
18
- else:
19
- return "negative"
20
 
21
 
22
 
 
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 tokenizer.decode(summarized_ids[0], skip_special_tokens=True)
16
+
 
 
 
17
 
18
 
19