nph4rd commited on
Commit
806a1a2
·
verified ·
1 Parent(s): eab0af2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -73,14 +73,16 @@ def parse_segmentation(input_image, input_text, model_choice):
73
 
74
  INTRO_TEXT = """## PaliGemma WaveUI\n\n
75
 
76
- PaliGemma models fine-tuned on the [WaveUI dataset](https://huggingface.co/datasets/agentsea/wave-ui). There are two base-models:\n\n
77
 
78
- - paligemma-3b-ft-widgetcap-waveui-448
79
- - paligemma-3b-ft-waveui-896
80
 
81
  Note:\n\n
82
 
83
- - the task it was fine-tuned on was detection, so it may not generalize to other tasks.
 
 
84
  """
85
 
86
  with gr.Blocks(css="style.css") as demo:
 
73
 
74
  INTRO_TEXT = """## PaliGemma WaveUI\n\n
75
 
76
+ Two fine-tuned models on the [WaveUI dataset](https://huggingface.co/datasets/agentsea/wave-ui) from different bases:\n\n
77
 
78
+ - [paligemma-3b-ft-widgetcap-waveui-448](https://huggingface.co/agentsea/paligemma-3b-ft-widgetcap-waveui-448)
79
+ - [paligemma-3b-ft-waveui-896](https://huggingface.co/agentsea/paligemma-3b-ft-waveui-896)
80
 
81
  Note:\n\n
82
 
83
+ - the task they were fine-tuned on was detection, so it may not generalize to other tasks.
84
+
85
+ Usage: write the task keyword "detect" before the element you want the model to detect. For example, "detect profile picture".
86
  """
87
 
88
  with gr.Blocks(css="style.css") as demo: