Update app.py

app.py

@@ -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 = model.generate(inputs, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
+outputs = fine_tuned_model.generate(inputs, max_length=100, temperature = .7, do_sample=True, pad_token_id=tokenizer.eos_token_id)
 
 completion = tokenizer.decode(outputs[0])