prompt

app.py

@@ -65,7 +65,7 @@ pipe.enable_vae_tiling()
 
 
 blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
-blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
+blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to(device)
 
 target_content_blocks = BLOCKS['content']
 target_style_blocks = BLOCKS['style']
@@ -184,8 +184,8 @@ def create_image(content_image_pil,
         content_image_pil = Image.fromarray(
             np.zeros((1024, 1024, 3), dtype=np.uint8)).convert('RGB')
 
-    if prompt is '':
-        blip_model = blip_model.to(device)
+    if prompt == '':
+
         inputs = blip_processor(content_image_pil, return_tensors="pt").to(device)
         out = blip_model.generate(**inputs)
         prompt = blip_processor.decode(out[0], skip_special_tokens=True)
@@ -324,6 +324,7 @@ with block:
 
                 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" and target == "Image-Driven Style Transfer":
                     prompt =''