BICORP commited on
Commit
8814186
·
verified ·
1 Parent(s): dc63802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -11,23 +11,24 @@ servers = [
11
  "BICORP/server-5"
12
  ]
13
 
14
- # Function to call the API with error handling
15
  def call_api(message, model, preset):
16
  selected_server = random.choice(servers)
17
  client = Client(selected_server)
18
 
19
  try:
 
20
  result = client.predict(
21
- message=message,
22
- param_2=model,
23
- param_3=preset,
24
  api_name="/chat"
25
  )
26
  return result
27
  except Exception as e:
28
  return f"Error: {str(e)}"
29
 
30
- # Custom CSS for modern UI
31
  css = """
32
  .gradio-container {
33
  background-color: #f9f9f9;
 
11
  "BICORP/server-5"
12
  ]
13
 
14
+ # Function to call the API with correct parameters
15
  def call_api(message, model, preset):
16
  selected_server = random.choice(servers)
17
  client = Client(selected_server)
18
 
19
  try:
20
+ # Get API parameter names (replace with actual API parameter names from "View API")
21
  result = client.predict(
22
+ message, # First input
23
+ model, # Second input (correct parameter name)
24
+ preset, # Third input (correct parameter name)
25
  api_name="/chat"
26
  )
27
  return result
28
  except Exception as e:
29
  return f"Error: {str(e)}"
30
 
31
+ # Custom CSS for clean UI
32
  css = """
33
  .gradio-container {
34
  background-color: #f9f9f9;