Update app.py

app.py

@@ -38,7 +38,7 @@ def generate_text(prompt, max_tokens=500, temperature=0.3, top_k=40):
             
             # Get predictions
             logits,_ = model(input_ids)
-            logits = outputs.logits[:, -1, :] / temperature
+            logits = logits[:, -1, :] / temperature
             
             # Apply top-k filtering
             if top_k > 0: