daniyal214 commited on
Commit
e904b51
·
1 Parent(s): 869cefd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import pipeline
3
 
4
  pipe = pipeline("image-to-text", model="daniyal214/finetuned-blip-chest-xrays")
5
 
6
- def get_captions(image_path):
7
- result = pipe("/content/PMC2740289_1757-1626-0002-0000006174-002.jpg")
8
  result = result[0]['generated_text']
9
  return result
10
 
 
3
 
4
  pipe = pipeline("image-to-text", model="daniyal214/finetuned-blip-chest-xrays")
5
 
6
+ def get_captions(image):
7
+ result = pipe(image)
8
  result = result[0]['generated_text']
9
  return result
10