Update app.py

app.py

@@ -242,7 +242,7 @@ if st.button("Run"):
             encoded_prompt = tokenizer(start, add_special_tokens=False, return_tensors="pt").input_ids
             encoded_prompt = encoded_prompt.to(trainer.model.device)
             # prediction
-            output_sequences = trainer.model.generate(
+            output_sequences = model.generate(
                                     input_ids=encoded_prompt,
                                     max_length=max_length,
                                     min_length=min_length,