Update app.py

app.py

@@ -3,7 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Load model and tokenizer from Hugging Face Hub
-model_id = "deepseek-ai/deepseek-coder-1.3b-base"
+# model_id = "deepseek-ai/deepseek-coder-1.3b-base"
+model_id = "gpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32)