Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 
-from transformers import AutoModel, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import TextGenerationPipeline 
 
 
 tokenizer = AutoTokenizer.from_pretrained("Sigurdur/icebreaker")
-model = AutoModel.from_pretrained("Sigurdur/icebreaker")
+model = AutoModelForCausalLM.from_pretrained("Sigurdur/icebreaker")
 pipeline = TextGenerationPipeline(model=model, tokenizer=tokenizer)
 
 def generate_text(inp):