Spaces:

Tonic
/

CheXRay

Running on Zero

Tonic commited on Jan 26

Commit

7a118e2

•

1 Parent(s): e36e5a2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,8 +27,8 @@ model = AutoModelForCausalLM.from_pretrained("StanfordAIMI/CheXagent-8b", torch_
 @spaces.GPU
 def generate(image, prompt):
-    if hasattr(image_input, "read"):
-        image = Image.open(io.BytesIO(image_input.read())).convert("RGB")
     else:
         image = image

 @spaces.GPU
 def generate(image, prompt):
+    if hasattr(image, "read"):
+        image = Image.open(io.BytesIO(image.read())).convert("RGB")
     else:
         image = image