Update app.py

app.py

@@ -40,9 +40,9 @@ def animate_img(encoded_string):
                 print(sd_video[0])
                 return sd_video[0]
     except:
-        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result1 = client1.predict(encoded_string, api_name="/resize_image")        
-        client = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        client = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result = client.predict(result1, 0, True, 1, 15, api_name="/video")
         return result[0]['video']
 
@@ -55,9 +55,12 @@ def create_video(prompt):
         with closing(create_connection(f"{url_sd3}", timeout=120)) as conn:
             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":""}}')
-            while True:
+            c = 0
+            while c < 45:
                 status = json.loads(conn.recv())['msg']
                 if status == 'estimation':
+                    c += 1
+                    time.sleep(1)
                     continue
                 if status == 'process_starts':
                     break
@@ -78,7 +81,7 @@ def create_video(prompt):
             conn.recv()
             conn.recv()
             photo = json.loads(conn.recv())['output']['data'][0]
-            photo = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
+            base64_string = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
 
             image_bytes = base64.b64decode(base64_string)
             with tempfile.NamedTemporaryFile(delete=False) as temp:
@@ -107,9 +110,9 @@ def create_video(prompt):
                 print(sd_video[0])
                 return sd_video[0]
     except:
-        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result1 = client1.predict(encoded_string, api_name="/resize_image")        
-        client = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        client = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result = client.predict(result1, 0, True, 1, 15, api_name="/video")
         return result[0]['video']