Update app.py

app.py

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