Moritz Stephan commited on
Commit
62a69ae
·
1 Parent(s): 34bd714

adding other methods to demo

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,8 +24,8 @@ 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 != "none" else None,
28
- "method": method if method != "none" else None,
29
  }
30
  response = requests.post(f"{ENDPOINT_URL}/completion", json=args)
31
  data = response.json()
@@ -66,7 +66,7 @@ demo = gr.Interface(
66
  gr.Radio(
67
  choices=[
68
  ("C3PO", "c3po"),
69
- ("DPO", "dpo"),
70
  ("SFT + Negatives", "sft_negatives"),
71
  ("SFT", "sft"),
72
  ("Baseline", "baseline")
 
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)
31
  data = response.json()
 
66
  gr.Radio(
67
  choices=[
68
  ("C3PO", "c3po"),
69
+ ("DPO", "dpo_after_sft"),
70
  ("SFT + Negatives", "sft_negatives"),
71
  ("SFT", "sft"),
72
  ("Baseline", "baseline")