Changed gpt2 to llama

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 model_name = "Glasshes/debate_v3.1_model"
-model = GPT2LMHeadModel.from_pretrained(model_name)
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
@@ -17,7 +17,7 @@ iface = gr.Interface(
     outputs="text",
     live=True,
     title="Debate Model",
-    description="This model generates text based on the input prompt.",
+    description="This model generates text based on the input prompt using Llama.",
 )
 
 iface.launch()