Update app.py

app.py

@@ -15,8 +15,8 @@ story = open('story.txt').read().replace("\n\n", "\n").replace("\n", " ").strip(
 
 def ask(question):
     inputs = tokenizer(story, question, truncation=True, padding=True)
-    prediction = torch.softmax(model(**inputs).logits, dim=-1).squeeze()
-    return {c: prediction[i].item() for c, i in CLASSES}
+    prediction = np.round(torch.softmax(model(**inputs).logits, 1).squeeze().numpy(), 3)
+    return {c: prediction[i] for c, i in CLASSES}
 
 
 demo = gr.Interface(
@@ -27,7 +27,7 @@ demo = gr.Interface(
     description="“ Albert and Dave find themselves on the pier. They go to a nearby restaurant where Albert orders "
                 "seagull meat. The waiter promptly serves Albert the meal. After taking a bite, he realizes "
                 "something. Albert pulls a gun out of his ruined jacket and shoots himself. ”\n\nWhy did Albert shoot "
-                "himself?\n\nCan you unravel the truth behind this epilogue by asking only yes/no questions?",
+                "himself?\n\nCan you unravel the truth behind this epilogue by asking only yes/no questions?\n\nPlease be specific about the time period you have in mind with your question.",
     article='Please refrain from embarrassing DeBERTa with dumb questions.\n\nCheck the repository for more detail: https://github.com/MrPio/The-Seagull-Story',
     examples=['Albert shoot himself for a reason',
               'Dave has a watch on his wrist',