Update app.py

app.py

@@ -1,5 +1,12 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
 fine_tuned_model = "ashioyajotham/results"
 
+# Load the tokenizer for the Falcon 7B model with remote code trust
+tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)
+
+# Set the padding token to be the same as the end-of-sequence token
+tokenizer.pad_token = tokenizer.eos_token
+
 def generate_code(input_text):
     inputs = tokenizer.encode(input_text, return_tensors='pt')
     # Use your fine-tuned model to generate code here