Update pages/Visualization.py

pages/Visualization.py

@@ -16,13 +16,10 @@ def generate_img(payload):
     response = requests.post(API_URL_img, headers=headers, json=payload)
     body = response.json()
     if 'error' in body:
-        if 'estimated_time' in body:
-            time.sleep(body['estimated_time'])
-        else:
-            print(body)
-            return
-        generate_img(payload)
-    return response.content
+        time.sleep(3)
+        response = requests.post(API_URL_img, headers=headers, json=payload)
+    else:       
+        return response.content
 
 
 st.markdown('# :female-student: Персональный помощник для студентов')