Update app.py

app.py

@@ -27,8 +27,8 @@ def predict_clothing(images):
     input_text = processor.apply_chat_template(messages, add_generation_prompt=True)
 
     output_texts = []
-    for url in images:
-        image = Image.open(requests.get(url, stream=True).raw)
+    for img_path in images:
+        image = Image.open(img_path)
         print(type(image))
         inputs = processor(image, input_text, add_special_tokens=False, return_tensors="pt").to(model.device)
         with torch.no_grad():
@@ -157,8 +157,9 @@ with gr.Blocks() as demo:
             use_input_image_size_as_output = gr.Checkbox(
                 label="use_input_image_size_as_output", info="Automatically adjust the output image size to be same as input image size. For editing and controlnet task, it can make sure the output image has the same size as input image leading to better performance", value=False,)
 
+    with gr.Row():
         # generate
-        generate_button = gr.Button("Generate Image")
+        generate_button = gr.Button("Try On!")
             
     with gr.Row():
         # output image