Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
-import requests
-from huggingface_hub import cached_download, hf_hub_url
+from gradio_client import Client
+import time
 
 
 MORE = """ ## TRY Other Models
@@ -11,24 +11,23 @@ MORE = """ ## TRY Other Models
 
 # Function
 def generate_image(prompt, base="Anime", motion="", step=8, progress=gr.Progress()):
-    API_URL = "https://api-inference.huggingface.co/models/KingNish/Instant-Video/generate_image_1"
-    headers = {"Authorization": f"Bearer {os.environ.get('HF_API_TOKEN')}"}  # Замените на ваш токен Hugging Face API
-
-    payload = {
-        "inputs": prompt,
-        "options": {
-            "base": base,
-            "motion": motion,
-            "step": step
-        }
-    }
-
-    response = requests.post(API_URL, headers=headers, json=payload)
-    response.raise_for_status()
-
-    video_url = response.json()[0]['video']
-    video_path = cached_download(video_url)
-    return video_path
+    retries = 3
+    for i in range(retries):
+        try:
+            result = client.predict(
+                prompt=prompt,
+                base=base,
+                motion=motion,
+                step=step,
+                api_name="/generate_image_1"
+            )
+            video_path = result["video"]
+            return video_path
+        except ValueError as e:
+            print(f"Ошибка подключения. Попытка {i+1} из {retries}.")
+            raise gr.Info(f"Ошибка подключения. Попытка {i+1} из {retries}.")
+            time.sleep(5)  # Пауза перед следующей попыткой
+    raise gr.Error("Не удалось подключиться к API. Повторите попытку позже.")
         
 # Gradio Interface
 with gr.Blocks(css="style.css") as demo: