prompt

app.py

@@ -184,7 +184,7 @@ def create_image(content_image_pil,
         content_image_pil = Image.fromarray(
             np.zeros((1024, 1024, 3), dtype=np.uint8)).convert('RGB')
 
-    if prompt is None:
+    if prompt is '':
         blip_model = blip_model.to(device)
         inputs = blip_processor(content_image_pil, return_tensors="pt").to(device)
         out = blip_model.generate(**inputs)
@@ -318,9 +318,14 @@ with block:
                                   value="Image-Driven Style Transfer",
                                   label="task")
 
+                prompt_type = gr.Radio(
+                    ["caption of Blip", "user input"],
+                    value="caption of Blip",
+                    label="task")
                 prompt = gr.Textbox(label="Prompt",
                                     value="there is a small house with a sheep statue on top of it")
-
+                if prompt_type == "caption of Blip":
+                    prompt =''
 
                 scale_c = gr.Slider(minimum=0, maximum=2.0, step=0.01, value=0.6, label="Content Scale")
                 scale_s = gr.Slider(minimum=0, maximum=2.0, step=0.01, value=1.0, label="Style Scale")