Update app.py

app.py

@@ -1,6 +1,6 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 fine_tuned_model = "ashioyajotham/results"
-
+model = AutoModelForCausalLM.from_pretrained(fine_tuned_model)
 # Load the tokenizer for the Falcon 7B model with remote code trust
 tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)
 
@@ -11,7 +11,7 @@ prompt = "Generate a Python script to add prime numbers between one and ten"
 
 inputs = tokenizer.encode(prompt, return_tensors='pt')
 
-outputs = fine_tuned_model.generate(inputs, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
+outputs = model.generate(inputs, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
 
 completion = tokenizer.decode(outputs[0])