Update app.py

app.py

@@ -1,20 +1,24 @@
 import gradio as gr
 from transformers import pipeline
-from transformers import AutoModelForCausalLM
 
-model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)
 messages = [
     {"role": "system", "content": "You are a chatbot that responds in 10 lines of shakespeare given a prompt"}
 ]
-pipe = pipeline("text-generation", model=model, trust_remote_code=True)
+pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
     
 def poet(input):
     messages.append(
     {"role": "user", "content": input})
-    return pipe(messages)
+    generated_text = pipe(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+    return generated_text
+    
 
-demo = gr.Interface(fn=greet, inputs="textbox", outputs="textbox")
+chat = gr.ChatInterface(
+    fn=generate_shakespeare,
+    title="Shakespearean Text Generator",
+    description="Enter a prompt and let the DeepSeek model generate Shakespearean-style text.",
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    chat.launch()