Update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import AutoModel, AutoTokenizer, pipeline
 import gradio as gr
 
 torch.random.manual_seed(0)
 
-model = AutoModelForCausalLM.from_pretrained(
+model = AutoModel.from_pretrained(
     "DuckyBlender/racist-phi3", 
     device_map="auto", 
     torch_dtype="auto", 
@@ -36,7 +36,7 @@ def phi3_fun(message,chat_history):
   messages=[
     {"role": "user", "content": message},
 ]
-  output = pipe(messages, **generation_args)
+  output = pipe(messages)
   respond = output[0]['generated_text']
   return respond