Update app.py

app.py

@@ -131,8 +131,13 @@ def my_inference_function(sec_txt):
 
     mean = get_mean_from_proba(proba_list)
 
-    return torch.argmax(mean).item()
-    
+    sentiment = torch.argmax(mean).item()
+    if sentiment == 0:
+        return 'Positive'
+    elif sentiment == 1:
+        return 'Negative'
+    else:
+        return 'Neutral'
     
 gradio_interface = gradio.Interface(
   fn = my_inference_function,