Update app.py

app.py

@@ -4,7 +4,8 @@ from flask import *
 from routes.helpers import configFile
 from routes import *
 
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+from random import randint
 
 #initing
 app = Flask(__name__)
@@ -129,7 +130,7 @@ def generateText():
     try:
         text = helpers.getFromRequest(request, "text")
         if not text: return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }} , 400
-        result = generator(text, max_length=1024, return_full_text=False, do_sample=bool(randint(0,1)), temperature=float(randint(5,15))/10.0)
+        result = generator(text, max_length=1024, return_full_text=False, do_sample=bool(randint(0,1)), temperature=float(randint(5,15))/10.0, max_time)
         return {"status": "pass", "result": result[0]["generated_text"]} 
     except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }} , 400