Update app.py

app.py

@@ -28,7 +28,7 @@ def animate_img(encoded_string):
             r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
             source_string = r2.text
             if "Generation has been in progress for" in source_string:
-                time.sleep(10)
+                time.sleep(15)
                 c += 1
                 continue
             if "Generation has been in progress for" not in source_string:
@@ -38,7 +38,10 @@ def animate_img(encoded_string):
                 for match in matches:
                     sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
                 print(sd_video[0])
-                return sd_video[0]
+                if len(sd_video) != 0:
+                    return sd_video[0]
+                else:
+                    _ = 1/0
     except:
         client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result1 = client1.predict(encoded_string, api_name="/resize_image")        
@@ -56,7 +59,7 @@ def create_video(prompt):
             conn.send('{"fn_index":3,"session_hash":""}')
             conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry",7.5,"(No style)"],"event_data":null,"fn_index":3,"session_hash":""}}')
             c = 0
-            while c < 45:
+            while c < 60:
                 status = json.loads(conn.recv())['msg']
                 if status == 'estimation':
                     c += 1
@@ -98,7 +101,7 @@ def create_video(prompt):
             r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
             source_string = r2.text
             if "Generation has been in progress for" in source_string:
-                time.sleep(10)
+                time.sleep(15)
                 c += 1
                 continue
             if "Generation has been in progress for" not in source_string:
@@ -108,7 +111,10 @@ def create_video(prompt):
                 for match in matches:
                     sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
                 print(sd_video[0])
-                return sd_video[0]
+                if len(sd_video) != 0:
+                    return sd_video[0]
+                else:
+                    _ = 1/0
     except:
         client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result1 = client1.predict(encoded_string, api_name="/resize_image")