Update app.py

app.py

@@ -11,7 +11,7 @@ login(token=token)
 
 
 # Load model and tokenizer
-model_name = "meta-llama/Llama-3.2-3B"  # Replace with the correct model name if necessary
+model_name = "nikunjcepatel/gpt_finetune_test"  #"meta-llama/Llama-3.2-3B"  # Replace with the correct model name if necessary
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)