Update app.py

app.py

@@ -48,19 +48,20 @@ def animate_img(encoded_string, model):
                         _ = 1/0
             print("f_1")
         except:
-            print("2")
-            client1 = Client(url_hg1)
-            result1 = client1.predict(encoded_string, api_name="/resize_image")        
-            client = Client(url_hg1)
-            result = client.predict(result1, 0, True, 1, 15, api_name="/video")
-            res = result[0]['video']
-            print("s_2")
-            return res
+            return None
+            #print("2")
+            #client1 = Client(url_hg1)
+            #result1 = client1.predict(encoded_string, api_name="/resize_image")        
+            #client = Client(url_hg1)
+            #result = client.predict(result1, 0, True, 1, 15, api_name="/video")
+            #res = result[0]['video']
+            #print("s_2")
+            #return res
 
-    if model == "AnimateDiff":
-        client = Client(url_hg2)
-        result = client.predict(encoded_string, "zoom-out", api_name="/predict")
-        return result
+    #if model == "AnimateDiff":
+    #    client = Client(url_hg2)
+    #    result = client.predict(encoded_string, "zoom-out", api_name="/predict")
+    #    return result
     
     
 def create_video(prompt, model):
@@ -84,12 +85,10 @@ def create_video(prompt, model):
                 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:
+                with tempfile.NamedTemporaryFile(delete=True) as temp:
                     temp.write(image_bytes)
                     temp_file_path = temp.name
                     print("cs_1")
-    
-                
         except:
             print("c_2")
             with closing(create_connection(f"{url_sd4}", timeout=120)) as conn:
@@ -103,7 +102,7 @@ def create_video(prompt, model):
                 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:
+                with tempfile.NamedTemporaryFile(delete=True) as temp:
                     temp.write(image_bytes)
                     temp_file_path = temp.name
                     print("cs_2")
@@ -133,11 +132,12 @@ def create_video(prompt, model):
                     else:
                         _ = 1/0
         except:
-            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")
-            result = client.predict(result1, 0, True, 1, 15, api_name="/video")
-            return result[0]['video']
+            return None
+            #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")
+            #result = client.predict(result1, 0, True, 1, 15, api_name="/video")
+            #return result[0]['video']
         
             
 
@@ -307,7 +307,7 @@ with gr.Blocks(css=css) as demo:
             prompt = gr.Textbox(placeholder="Введите описание видео...", show_label=True, label='Описание:', lines=3)
             # motion1 = gr.Dropdown(value="Приближение →←", interactive=True, show_label=True, label="Движение камеры:", choices=["Приближение →←", "Отдаление ←→", "Вверх ↑", "Вниз ↓", "Влево ←", "Вправо →", "По часовой стрелке ⟳", "Против часовой стрелки ⟲"])
             model = gr.Radio(interactive=True, value="Stable Video Diffusion", show_label=True, 
-                             label="Модель нейросети:", choices=['Stable Video Diffusion', 'AnimateDiff'])
+                             label="Модель нейросети:", choices=['Stable Video Diffusion'])
         with gr.Column():
             text_button = gr.Button("Сгенерировать видео", variant='primary', elem_id="generate")
         with gr.Column():