Upload app.py

app.py

@@ -217,7 +217,7 @@ except Exception as e:
 model.to(device)
 model.eval()
 
-def generate_text(prompt, max_length=100, temperature=0.7, top_k=50):
+def generate_text(prompt, max_length=20, temperature=0.7, top_k=50):
     """Generate text from a prompt"""
     input_ids = tokenizer.encode(prompt, return_tensors='pt').to(device)