Sree

app.py

@@ -1,12 +1,30 @@
+#import gradio as gr
+
+#description = "Story generation with GPT-2"
+#title = "Generate your own story"
+#examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
+
+#interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
+#            description=description,
+#            examples=examples
+#)
+
+#interface.launch()
+
 import gradio as gr
 
+#gr.Interface.load("models/NousResearch/Yarn-Mistral-7b-128k").launch()
+
 description = "Story generation with GPT-2"
 title = "Generate your own story"
 examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
 
-interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
+#interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
+
+gr.Interface.load("models/NousResearch/Yarn-Mistral-7b-128k",
             description=description,
             examples=examples
 )
 
 interface.launch()
+