Update app.py

app.py

@@ -12,7 +12,11 @@ def summarize(text):
     summarized_ids = model.generate(inputs['input_ids'], attention_mask=inputs['attention_mask'], 
                                     max_length=150, num_beams=4, early_stopping=True)
 
-    return tokenizer.decode(summarized_ids[0], skip_special_tokens=True)
+    f= tokenizer.decode(summarized_ids[0], skip_special_tokens=True)
+    if f == 1:
+        return "positive"
+    else:
+        return "negative"