Update app.py

app.py

@@ -47,9 +47,9 @@ prompt = PromptTemplate(
 )
 
 # === Load Falcon Model ===
-model_name = "mistralai/Mistral-7B-Instruct-v0.2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+model = AutoModelForCausalLM.from_pretrained("gpt2").to("cuda" if torch.cuda.is_available() else "cpu")
+
 
 generator = pipeline(
     "text-generation",