Update app.py

app.py

@@ -25,18 +25,16 @@ def infer(prompt, max_new_tokens=10, temperature=0.0, top_p=1.0):
         "source": "dalle",
     }
     
-    res = requests.post("https://planetd.shift.ml/jobs", json=my_post_dict).json()
-    
-    job_id = res['id']
+    job_id = requests.post("https://planetd.shift.ml/jobs", json=my_post_dict).json()['id']
     
     while True:
+    
+        time.sleep(1)
         
         ret = requests.get(f"https://planetd.shift.ml/job/{job_id}", json={'id': job_id}).json()
         
         if ret['status'] == 'finished':
             break
-  
-        time.sleep(1)
         
     return ret['returned_payload']['result']['inference_result'][0]['choices'][0]['text']
     
@@ -51,15 +49,14 @@ prompt = st.text_area(
     height=400,
 )
     
-    
 generated_area = st.empty()
 generated_area.markdown("(Generate here)")
 
 button_submit = st.button("Submit")
    
-max_new_tokens = st.number_input('Max new tokens', 1, 1024, 10)
-temperature = st.number_input('temperature', 0.0, 10.0, 0.0, step=0.1, format="%.2f")
-top_p = st.number_input('top_p', 0.0, 1.0, 1.0, step=0.1, format="%.2f")
+max_new_tokens = st.text_input('Max new tokens', "10")
+temperature = st.text_input('temperature', "0.0")
+top_p = st.text_input('top_p', "1.0")
 
 if button_submit:
     with st.spinner(text="In progress.."):