Update app.py

app.py

@@ -7,7 +7,6 @@ import json
 import time
 
 API_URL = os.environ.get("API_URL")
-LEPTON_API_TOKEN = os.environ.get('LEPTON_API_TOKEN')
 
 supported_languages = ['zh', 'en']
 
@@ -104,12 +103,8 @@ def predict(prompt, style, audio_file_pth, agree):
     }
     
     start = time.time()
-    # # Send the data as a POST request
-    # response = requests.post(API_URL, json=data, timeout=60)
-    
-    from leptonai.client import Client
-    c = Client(API_URL, token=LEPTON_API_TOKEN)
-    response = c.run(**data)
+    # Send the data as a POST request
+    response = requests.post(API_URL, json=data, timeout=60)
     
     print(f'Get response successfully within {time.time() - start}')
 
@@ -128,32 +123,32 @@ def predict(prompt, style, audio_file_pth, agree):
         with open(save_path, 'wb') as f:
             f.write(response)
     
-    # # Check the response
-    # if response.status_code == 200:
-    #     try:
-    #         json_data = json.loads(response.content)
-    #         text_hint += f"[ERROR] {json_data['error']} \n"
-    #         gr.Warning(
-    #             f"[ERROR] {json_data['error']} \n"
-    #         )
-    #         return (
-    #             text_hint,
-    #             None,
-    #             None,
-    #         )
-    #     except:
-    #         with open(save_path, 'wb') as f:
-    #             f.write(response.content)
-    # else:
-    #     text_hint += f"[HTTP ERROR] {response.status_code} - {response.text} \n"
-    #     gr.Warning(
-    #         f"[HTTP ERROR] {response.status_code} - {response.text} \n"
-    #     )
-    #     return (
-    #         text_hint,
-    #         None,
-    #         None,
-    #     )
+    # Check the response
+    if response.status_code == 200:
+        try:
+            json_data = json.loads(response.content)
+            text_hint += f"[ERROR] {json_data['error']} \n"
+            gr.Warning(
+                f"[ERROR] {json_data['error']} \n"
+            )
+            return (
+                text_hint,
+                None,
+                None,
+            )
+        except:
+            with open(save_path, 'wb') as f:
+                f.write(response.content)
+    else:
+        text_hint += f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+        gr.Warning(
+            f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+        )
+        return (
+            text_hint,
+            None,
+            None,
+        )
             
     text_hint += f'''Get response successfully \n'''
     return (