Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-generator = pipeline("text-generation", "Ayham/albert_gpt2_Full_summarization_cnndm")
+pipe = pipeline("text-generation","TheBloke/Mistral-7B-Instruct-v0.1-GGUF")
 
-def generate(text):
-    output = generator(text)
-    return output[0]['generated_text']
+def chat_fn(message, history):
+    response = pipe(message, max_new_tokens=200)[0]["generated_text"]
+    return response
+
+gr.ChatInterface(fn=chat_fn).launch()
 
-iface = gr.Interface(fn=generate, inputs="text", outputs="text", title="FlareNew Text Generation")
-iface.launch()