joaogante HF staff commited on
Commit
e6473c1
·
1 Parent(s): b258678

add basic logic

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -33,13 +33,13 @@ assert all(tag in HUB_TAGS for tag in TASK_TYPES.values())
33
 
34
  # Dict with the problems considered in this spaces, {problem: space tag}
35
  PROBLEMS = {
36
- "🌈 I would like a ChatGPT-like model": "chatgpt",
37
- "📈 I want to improve the overall quality of the output": "quality",
38
- "🏎 Speed! Make it faster!": "faster",
39
- "😵‍💫 I would like to reduce model hallucinations": "hallucinations",
40
- "📏 I want to control the length of the output": "length",
41
- "🤔 The model is returning nothing / random words": "random",
42
- "❓ Other": "other",
43
  }
44
 
45
  INIT_MARKDOWN = """
@@ -115,7 +115,7 @@ with demo:
115
  interactive=True,
116
  )
117
  problem_type = gr.Dropdown(
118
- label="What would you like to improve?",
119
  choices=list(PROBLEMS.keys()),
120
  interactive=True,
121
  )
 
33
 
34
  # Dict with the problems considered in this spaces, {problem: space tag}
35
  PROBLEMS = {
36
+ "🤔 Baseline. The model is returning nothing / random words": "baseline",
37
+ "😵‍💫 Hallucinations. I would like to reduce them": "hallucinations",
38
+ "🌈 Interactivity. I would like a ChatGPT-like model": "interactity",
39
+ "📏 Length. I want to control the length of the output": "length",
40
+ "📈 Quality. I want to improve the overall quality": "quality",
41
+ "🏎 Speed! Make it faster!": "speed",
42
+ "❓ ??? Something else, looking for ideas": "other",
43
  }
44
 
45
  INIT_MARKDOWN = """
 
115
  interactive=True,
116
  )
117
  problem_type = gr.Dropdown(
118
+ label="What would you like to improve? (sorted by keyword)",
119
  choices=list(PROBLEMS.keys()),
120
  interactive=True,
121
  )