Update app.py

app.py

@@ -1,11 +1,23 @@
-import gradio as gr
-print("hello")
+# import gradio as gr
+# print("hello")
 # gr.load("NousResearch/Yarn-Mistral-7b-128k",src="models").launch()
+# from transformers import pipeline
+
+# classifier = pipeline("sentiment-analysis")
+# def get_class(input):
+#     return classifier("I've been waiting for a HuggingFace course my whole life.")
+
+# iface = gr.Interface(fn=get_class, inputs="text", outputs=['text'], title='hello', description='play around')
+# iface.launch()
+
+
+# Use a pipeline as a high-level helper
 from transformers import pipeline
 
-classifier = pipeline("sentiment-analysis")
+pipe = pipeline("text-generation", model="NousResearch/Yarn-Mistral-7b-128k")
+
 def get_class(input):
-    return classifier("I've been waiting for a HuggingFace course my whole life.")
+    return pipe(input)
 
 iface = gr.Interface(fn=get_class, inputs="text", outputs=['text'], title='hello', description='play around')
-iface.launch()
+iface.launch()