daniyal214 commited on
Commit
46809cd
1 Parent(s): be91a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
  from PIL import Image
4
  import numpy as np
5
 
6
- pipe = pipeline("image-to-text", model="daniyal214/finetuned-blip-chest-xrays")
7
 
8
  def get_captions(input_image):
9
  # Convert the received image to a PIL Image
 
3
  from PIL import Image
4
  import numpy as np
5
 
6
+ pipe = pipeline("image-to-text", model="daniyal214/finetuned-git-large-chest-xrays")
7
 
8
  def get_captions(input_image):
9
  # Convert the received image to a PIL Image