Upload 3 files

app.py

@@ -13,6 +13,6 @@ def home():
 @app.get("/generate")
 def generate(text: str):
     # Use the pipeline to generate text from the given input text
-    output = pipe(text)
+    output = pipe(text,max_new_tokens=100)
     # Return the generated text in JSON response
     return {"output": output[0]['generated_text']}