Sree

app.py

@@ -15,16 +15,23 @@ 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."]]
+#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",
 
-gr.Interface.load("models/NousResearch/Yarn-Mistral-7b-128k",
-            description=description,
-            examples=examples
-)
+#gr.Interface.load("models/NousResearch/Yarn-Mistral-7b-128k",
+#            description=description,
+#            examples=examples
+#)
+
+#interface.launch()
+
+import gradio as gr
 
-interface.launch()
+def image_classifier(inp):
+    return {'cat': 0.3, 'dog': 0.7}
 
+demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
+demo.launch()