Moritz Stephan commited on
Commit
1f7ec24
·
1 Parent(s): dd27d00

adding disable adapter checkbox

Browse files
Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -20,10 +20,10 @@ def get_feedback_options() -> List[Tuple[str, str]]:
20
  for adapter in data]
21
 
22
 
23
- def get_completion(prompt: str, adapter: Optional[str] = None) -> str:
24
  args = {
25
  "C3PO_API_KEY": os.environ.get("C3PO_API_KEY"),
26
- "prompt": prompt,
27
  "adapter": adapter,
28
  }
29
  response = requests.post(f"{ENDPOINT_URL}/completion", json=args)
@@ -43,11 +43,23 @@ dropdown_options = get_feedback_options()
43
  demo = gr.Interface(
44
  get_completion,
45
  [
 
 
 
 
 
 
 
 
 
46
  gr.Textbox(
47
  placeholder="Enter a prompt...", label="Prompt"
48
  ),
49
  gr.Dropdown(
50
  choices=dropdown_options, label="Feedback", info="Will add the adapter for the respective feedback to the model."
 
 
 
51
  )
52
  ],
53
  "text",
 
20
  for adapter in data]
21
 
22
 
23
+ def get_completion(prompt: str, adapter: Optional[str], enable_feedback: bool) -> str:
24
  args = {
25
  "C3PO_API_KEY": os.environ.get("C3PO_API_KEY"),
26
+ "prompt": prompt if enable_feedback else None,
27
  "adapter": adapter,
28
  }
29
  response = requests.post(f"{ENDPOINT_URL}/completion", json=args)
 
43
  demo = gr.Interface(
44
  get_completion,
45
  [
46
+ gr.Markdown(
47
+ """
48
+ # C3PO Demo
49
+
50
+ This is a demo of Contextualized Critiques with Constrained Preference Optimization (C3PO). See the project website [here](<insert link>), repo [here](<insert link>), and the paper [here](<insert link>).
51
+
52
+ Selecting a feedback in the dropdown and enabling the "Use Feedback Adapter" checkbox will add the respective adapter to the model. The model will then use the feedback to generate the completion.
53
+ """
54
+ ),
55
  gr.Textbox(
56
  placeholder="Enter a prompt...", label="Prompt"
57
  ),
58
  gr.Dropdown(
59
  choices=dropdown_options, label="Feedback", info="Will add the adapter for the respective feedback to the model."
60
+ ),
61
+ gr.Checkbox(
62
+ label="Use Feedback Adapter", default=True, info="Enables the feedback adapter."
63
  )
64
  ],
65
  "text",