gitlost-murali commited on
Commit
08a2a10
1 Parent(s): 7599536

add resize checkbox

Browse files
Files changed (2) hide show
  1. app.py +4 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -7,9 +7,10 @@ def main():
7
  # Gradio Interface
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."
 
7
  # Gradio Interface
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
+ gr.Checkbox(label="Resize (Recommended)", value=True, info="Resize the image to 1920x1080 for better results?")],
14
  outputs=gr.Image(type="pil"),
15
  title="PTA-Text: A **Text-only** Click Model",
16
  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."
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- askui-ml-helper>=0.1.3
2
  gradio
 
1
+ askui-ml-helper>=0.1.4
2
  gradio