Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import GPTForCausalLM, AutoTokenizer, pipeline
 
 # Use a pipeline as a high-level helper for text generation
-pipe = pipeline("text-generation", model="Wonder-Griffin/TraXLMistral")
+pipe = pipeline("text-classification", model="Wonder-Griffin/JudgeLLM2")
 
 # Assuming `model_path` is the Hugging Face model hub path or a local directory
-model_path = "Wonder-Griffin/TraXLMistral"  # Define this as needed
+model_path = "Wonder-Griffin/JudgeLLM2"  # Define this as needed
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
+model = GPTForCausalLM.from_pretrained(
     model_path,
     device_map="auto",
     torch_dtype='auto'