Update app.py

app.py

@@ -3,11 +3,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import spaces
 
 # Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("Yoxas/autotrain-phi3-statistical", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Yoxas/autotrain-phi3-statistical", trust_remote_code=True)
+
+tokenizer = AutoTokenizer.from_pretrained("Yoxas/autotrain-gpt2-statistical1")
+model = AutoModelForCausalLM.from_pretrained("Yoxas/autotrain-gpt2-statistical1")
 
 # Use a pipeline as a high-level helper
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code=True)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 # Define the chatbot function
 @spaces.GPU(duration=120)