gitlost-murali commited on
Commit
7599536
·
1 Parent(s): e808dea

add prediction dot size

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ def main():
8
  iface = gr.Interface(
9
  fn=inference.process_image_and_draw_circle,
10
  inputs=[gr.Image(type="pil", label = "Upload Image"),
11
- gr.Textbox(label = "Prompt", placeholder='Enter a prompt to find in the image', value='Click on the text "Login"')],
 
12
  outputs=gr.Image(type="pil"),
13
  title="PTA-Text: A **Text-only** Click Model",
14
  description="Based on PTA-text ([Huggingface model card](https://huggingface.co/AskUI/pta-text-0.1) ).PTA is a small part of our bigger model - PTA-1 (Prompt-To-Automation). \nInstructions: Upload a UI image and enter a text finding prompt to see the model's prediction. \nLimitations: Currently, limited to single screen images."
 
8
  iface = gr.Interface(
9
  fn=inference.process_image_and_draw_circle,
10
  inputs=[gr.Image(type="pil", label = "Upload Image"),
11
+ gr.Textbox(label = "Prompt", placeholder='Enter a prompt to find in the image', value='Click on the text "Login"'),
12
+ gr.Slider(minimum=5, maximum=20, value=10, label="Prediction Dot Size", step=5)],
13
  outputs=gr.Image(type="pil"),
14
  title="PTA-Text: A **Text-only** Click Model",
15
  description="Based on PTA-text ([Huggingface model card](https://huggingface.co/AskUI/pta-text-0.1) ).PTA is a small part of our bigger model - PTA-1 (Prompt-To-Automation). \nInstructions: Upload a UI image and enter a text finding prompt to see the model's prediction. \nLimitations: Currently, limited to single screen images."