hubsnippetai commited on
Commit
75d3ae5
·
verified ·
1 Parent(s): 48935ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
9
 
10
  # The original Kosmos-2 demo saves the image first then reload it. For some images, this will give slightly different image input and change the generation outputs.
11
 
12
- prompt = "If the image is related to a medical or a health image start with 'An image of', otherwise 'The image is not related to health or medical'"
13
 
14
  def describe_image(image_path):
15
  inputs = processor(text=prompt, images=image_path, return_tensors="pt")
 
9
 
10
  # The original Kosmos-2 demo saves the image first then reload it. For some images, this will give slightly different image input and change the generation outputs.
11
 
12
+ prompt = ""
13
 
14
  def describe_image(image_path):
15
  inputs = processor(text=prompt, images=image_path, return_tensors="pt")