Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,19 +12,21 @@ servers = [
|
|
12 |
"BICORP/server-6"
|
13 |
]
|
14 |
|
15 |
-
# Function to call the API
|
16 |
def call_api(message, model, preset):
|
17 |
selected_server = random.choice(servers)
|
18 |
client = Client(selected_server)
|
19 |
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
|
|
|
|
28 |
|
29 |
# Custom CSS for styling
|
30 |
css = """
|
@@ -90,7 +92,7 @@ def create_interface():
|
|
90 |
model = gr.Dropdown(choices=["Lake 1 Base"], label="Select Model", value="Lake 1 Base")
|
91 |
preset = gr.Dropdown(choices=["Fast", "Normal", "Quality", "Unreal Performance"], label="Select Preset", value="Fast")
|
92 |
|
93 |
-
# Button click event
|
94 |
submit_btn.click(call_api, inputs=[message, model, preset], outputs=output)
|
95 |
|
96 |
return demo
|
|
|
12 |
"BICORP/server-6"
|
13 |
]
|
14 |
|
15 |
+
# Function to call the API with error handling
|
16 |
def call_api(message, model, preset):
|
17 |
selected_server = random.choice(servers)
|
18 |
client = Client(selected_server)
|
19 |
|
20 |
+
try:
|
21 |
+
result = client.predict(
|
22 |
+
message=message,
|
23 |
+
param_2=model,
|
24 |
+
param_3=preset,
|
25 |
+
api_name="/chat"
|
26 |
+
)
|
27 |
+
return result
|
28 |
+
except Exception as e:
|
29 |
+
return f"Error: {str(e)}"
|
30 |
|
31 |
# Custom CSS for styling
|
32 |
css = """
|
|
|
92 |
model = gr.Dropdown(choices=["Lake 1 Base"], label="Select Model", value="Lake 1 Base")
|
93 |
preset = gr.Dropdown(choices=["Fast", "Normal", "Quality", "Unreal Performance"], label="Select Preset", value="Fast")
|
94 |
|
95 |
+
# Button click event with loading indicator
|
96 |
submit_btn.click(call_api, inputs=[message, model, preset], outputs=output)
|
97 |
|
98 |
return demo
|