Setup Gradio space

app.py

@@ -7,9 +7,9 @@ CLASSES = {
     'irrelevant': 1,
     'no': 2,
 }
+tokenizer = AutoTokenizer.from_pretrained('cross-encoder/nli-deberta-v3-base',do_lower_case=True)
 model = AutoModelForSequenceClassification.from_pretrained('MrPio/TheSeagullStory-nli-deberta-v3-base')
 model.eval()
-tokenizer = AutoTokenizer.from_pretrained('cross-encoder/nli-deberta-v3-base')
 story = open('story.txt').read().replace("\n\n", "\n").replace("\n", " ").strip()
 
 
@@ -24,8 +24,11 @@ demo = gr.Interface(
     inputs=[gr.Textbox(value="", label="Your question, as an affirmative sentence:")],
     outputs=[gr.Label(label="Answer", num_top_classes=3)],
     title="The Seagull Story",
-    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?",
-    article='Please refrain from embarrassing DeBERTa with ridiculous questions.',
+    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?",
+    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',
               'Albert and Dave came to the pier on their own']

requirements.txt

@@ -1,4 +1,5 @@
 huggingface_hub==0.25.2
 transformers
+tokenizers
 torch
 gradio