Update app.py

app.py

@@ -1,18 +1,7 @@
+from transformers import pipeline
 import gradio as gr
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from transformers import TextGenerationPipeline 
+pipe = pipeline("text-generation", model="Sigurdur/icebreaker")
 
-
-tokenizer = AutoTokenizer.from_pretrained("Sigurdur/icebreaker")
-model = AutoModelForCausalLM.from_pretrained("Sigurdur/icebreaker")
-pipeline = TextGenerationPipeline(model=model, tokenizer=tokenizer)
-
-def generate_text(inp):
-    generated_text = pipeline(inp)
-
-    output = generated_text[0]["generated_text"]
-    return output
-
-output_text = gr.outputs.Textbox()
-gr.Interface(generate_text,"textbox",output_text,title="Text Generation machine ",description="Ask any question. Note: It can take 20-60 seconds to generate output based on your internet connection.").launch()
+demo = gr.Interface.from_pipeline(pipe)
+demo.launch()