Update app.py

app.py

@@ -7,7 +7,7 @@ def infer(prompt, max_new_tokens=10, temperature=0.0, top_p=1.0):
     my_post_dict = {
         "type": "general",
         "payload": {
-            "max_tokens": max_new_tokens,
+            "max_tokens": int(max_new_tokens),
             "n": 1,
             "temperature": float(temperature),
             "top_p": float(top_p),
@@ -27,7 +27,7 @@ def infer(prompt, max_new_tokens=10, temperature=0.0, top_p=1.0):
     
     job_id = requests.post("https://planetd.shift.ml/jobs", json=my_post_dict).json()['id']
     
-    while True:
+    for i in range(100):
     
         time.sleep(1)