Update app.py

app.py

@@ -90,21 +90,24 @@ def create_video(prompt, model):
                 #    temp_file_path = temp.name
                 #    print("cs_1")
         except:
-            print("c_2")
-            with closing(create_connection(f"{url_sd4}", timeout=120)) as conn:
-                conn.send('{"fn_index":0,"session_hash":""}')
-                conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                conn.recv()
-                photo = json.loads(conn.recv())['output']['data'][0]
-                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:
-                #    temp.write(image_bytes)
-                #    temp_file_path = temp.name
-                #    print("cs_2")
+            try:
+                print("c_2")
+                with closing(create_connection(f"{url_sd4}", timeout=120)) as conn:
+                    conn.send('{"fn_index":0,"session_hash":""}')
+                    conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
+                    conn.recv()
+                    conn.recv()
+                    conn.recv()
+                    conn.recv()
+                    photo = json.loads(conn.recv())['output']['data'][0]
+                    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:
+                    #    temp.write(image_bytes)
+                    #    temp_file_path = temp.name
+                    #    print("cs_2")
+            except:
+                return None
     
         try:
             r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": image_bytes})