Spaces:

daniyal214
/

gradio-caption-generator-git-large

Sleeping

daniyal214 commited on Jul 27, 2023

Commit

46809cd

1 Parent(s): be91a82

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 from PIL import Image
 import numpy as np
-pipe = pipeline("image-to-text", model="daniyal214/finetuned-blip-chest-xrays")
 def get_captions(input_image):
     # Convert the received image to a PIL Image

 from PIL import Image
 import numpy as np
+pipe = pipeline("image-to-text", model="daniyal214/finetuned-git-large-chest-xrays")
 def get_captions(input_image):
     # Convert the received image to a PIL Image