BICORP commited on
Commit
7b1926c
·
verified ·
1 Parent(s): c6399c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
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
- result = client.predict(
21
- message=message,
22
- param_2=model,
23
- param_3=preset,
24
- api_name="/chat"
25
- )
26
-
27
- return result
 
 
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