Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import GPTForCasualLM, AutoTokenizer, pipeline
+from transformers import AutoModelForCasualLM, AutoTokenizer, pipeline
 
 # Use a pipeline as a high-level helper for text generation
 pipe = pipeline("text-classification", model="Wonder-Griffin/JudgeLLM2")
@@ -8,7 +8,7 @@ pipe = pipeline("text-classification", model="Wonder-Griffin/JudgeLLM2")
 model_path = "Wonder-Griffin/JudgeLLM2"  # Define this as needed
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = GPTForCasualLM.from_pretrained(
+model = AutoModelForCasualLM.from_pretrained(
     model_path,
     device_map="auto",
     torch_dtype='auto'