Moritz Stephan commited on
Commit
da952a9
·
1 Parent(s): 215222e

added multi feedback support

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -20,11 +20,11 @@ def get_feedback_options() -> List[Tuple[str, str]]:
20
  for adapter in data]
21
 
22
 
23
- def get_completion(_, prompt: str, adapter: Optional[str], method: str) -> str:
24
  args = {
25
  "C3PO_API_KEY": os.environ.get("C3PO_API_KEY"),
26
  "prompt": prompt,
27
- "adapter": adapter if method != "baseline" else None,
28
  "method": method if method != "baseline" else None,
29
  }
30
  response = requests.post(f"{ENDPOINT_URL}/completion", json=args)
@@ -61,7 +61,9 @@ demo = gr.Interface(
61
  ),
62
  gr.Dropdown(
63
  choices=dropdown_options, label="Feedback", info="Will add the adapter for the respective feedback to the model.",
64
- value=dropdown_options[0][1]
 
 
65
  ),
66
  gr.Radio(
67
  choices=[
 
20
  for adapter in data]
21
 
22
 
23
+ def get_completion(_, prompt: str, adapters: Optional[list[str]], method: str) -> str:
24
  args = {
25
  "C3PO_API_KEY": os.environ.get("C3PO_API_KEY"),
26
  "prompt": prompt,
27
+ "adapters": adapters if method != "baseline" else None,
28
  "method": method if method != "baseline" else None,
29
  }
30
  response = requests.post(f"{ENDPOINT_URL}/completion", json=args)
 
61
  ),
62
  gr.Dropdown(
63
  choices=dropdown_options, label="Feedback", info="Will add the adapter for the respective feedback to the model.",
64
+ value=dropdown_options[0][1],
65
+ multiselect=True,
66
+ max_choices=3
67
  ),
68
  gr.Radio(
69
  choices=[