Update app.py

app.py

@@ -1,10 +1,12 @@
-from transformers import GPT2LMHeadModel, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
 import time
 
-model = GPT2LMHeadModel.from_pretrained("GBhaveshKumar/ConvoAI")
-tokenizer = AutoTokenizer.from_pretrained("GBhaveshKumar/ConvoAI")
+# Load model directly
+
+model = AutoModelForCausalLM.from_pretrained("Wonder-Griffin/ZeusForCausalLM", trust_remote_code=True, torch_dtype="auto"),)
+tokenizer = AutoTokenizer.from_pretrained("Wonder-Griffin/ZeusForCausalLM")
 
 device = torch.device("cpu") 
 model = model.to(device)