Spaces:
Runtime error
Runtime error
conditional
Browse files
app.py
CHANGED
@@ -7,9 +7,10 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
|
|
7 |
|
8 |
file_name = st.file_uploader("Upload image 1")
|
9 |
|
10 |
-
|
|
|
11 |
|
12 |
-
inputs = processor(image, return_tensors="pt")
|
13 |
|
14 |
-
out = model.generate(**inputs)
|
15 |
-
st.write(processor.decode(out[0], skip_special_tokens=True))
|
|
|
7 |
|
8 |
file_name = st.file_uploader("Upload image 1")
|
9 |
|
10 |
+
if file_name is not None:
|
11 |
+
image = Image.open(file_name)
|
12 |
|
13 |
+
inputs = processor(image, return_tensors="pt")
|
14 |
|
15 |
+
out = model.generate(**inputs)
|
16 |
+
st.write(processor.decode(out[0], skip_special_tokens=True))
|