Update app.py

app.py

@@ -67,16 +67,14 @@ def generate_text(prompt, max_tokens=50, temperature=0.8):
                 probs = torch.nn.functional.softmax(logits, dim=-1)
                 next_token = torch.multinomial(probs, num_samples=1)
                 x = torch.cat([x, next_token], dim=1)
-                
-                if next_token.item() == tokenizer.eos_token_id:
-                    break
         
         output_tokens = x[0].tolist()
         output = tokenizer.decode(output_tokens)
         return output
         
     except Exception as e:
-        return f"Error: {str(e)}"
+        import traceback
+        return f"Error: {str(e)}\n\n{traceback.format_exc()}"
 
 demo = gr.Interface(
     fn=generate_text,