Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_directory = "./"
 
 # Load the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained(model_directory)
+model = AutoModelForCausalLM.from_pretrained(model_directory, from_tf=True)
 tokenizer = AutoTokenizer.from_pretrained(model_directory, trust_remote_code=True)
 
 # Now you can generate text as before