Update app.py

app.py

@@ -18,6 +18,9 @@ from gradio_client import Client
 
 
 def animate_img(encoded_string, model):
+    url_hg1 = os.getenv("url_hg1")
+    url_hg2 = os.getenv("url_hg2")
+    
     if model == "Stable Video Diffusion":
         try:   
             r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(encoded_string, 'rb')})
@@ -45,16 +48,16 @@ def animate_img(encoded_string, model):
             print("f_1")
         except:
             print("2")
-            client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
+            client1 = Client(url_hg1)
             result1 = client1.predict(encoded_string, api_name="/resize_image")        
-            client = Client("https://emmadrex-stable-video-diffusion.hf.space")
+            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("https://ap123-animateimage.hf.space")
+        client = Client(url_hg2)
         result = client.predict(encoded_string, "zoom-out", api_name="/predict")
         return result
     
@@ -303,7 +306,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Tab("Анимировать изображение"):
         with gr.Column():
             prompt2 = gr.Image(show_label=True, interactive=True, type='filepath', label='Исходное изображение:')
-            prompt12 = gr.Textbox(placeholder="Введите описание видео...", show_label=True, label='Описание видео (опционально):', lines=3)
+            # prompt12 = gr.Textbox(placeholder="Введите описание видео...", show_label=True, label='Описание видео (опционально):', lines=3)
             # motion2 = gr.Dropdown(value="Приближение →←", interactive=True, show_label=True, label="Движение камеры:", choices=["Приближение →←", "Отдаление ←→", "Вверх ↑", "Вниз ↓", "Влево ←", "Вправо →", "По часовой стрелке ⟳", "Против часовой стрелки ⟲"])
             model2 = gr.Radio(interactive=True, value="Stable Video Diffusion", show_label=True, 
                              label="Модель нейросети:", choices=['Stable Video Diffusion', 'AnimateDiff'])