Spaces:
Sleeping
Sleeping
Commit
·
f99ea1d
1
Parent(s):
d5f26b3
Update pages/Captionize.py
Browse files- pages/Captionize.py +4 -21
pages/Captionize.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import torch
|
| 2 |
import re
|
|
|
|
| 3 |
from PIL import Image
|
| 4 |
import requests
|
| 5 |
import streamlit as st
|
|
@@ -31,28 +32,10 @@ div.stButton > button:hover {
|
|
| 31 |
}
|
| 32 |
</style>""", unsafe_allow_html=True)
|
| 33 |
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
list_image_urls = [
|
| 37 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/5/5a/Hyla_japonica_sep01.jpg/260px-Hyla_japonica_sep01.jpg",
|
| 38 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/7/71/Tibur%C3%B3n_azul_%28Prionace_glauca%29%2C_canal_Fayal-Pico%2C_islas_Azores%2C_Portugal%2C_2020-07-27%2C_DD_14.jpg/270px-Tibur%C3%B3n_azul_%28Prionace_glauca%29%2C_canal_Fayal-Pico%2C_islas_Azores%2C_Portugal%2C_2020-07-27%2C_DD_14.jpg",
|
| 39 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/2/21/Thure_de_Thulstrup_-_Battle_of_Shiloh.jpg/251px-Thure_de_Thulstrup_-_Battle_of_Shiloh.jpg",
|
| 40 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/2/21/Passion_fruits_-_whole_and_halved.jpg/270px-Passion_fruits_-_whole_and_halved.jpg",
|
| 41 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/5/5e/Messier83_-_Heic1403a.jpg/277px-Messier83_-_Heic1403a.jpg",
|
| 42 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/b/b6/2022-01-22_Men%27s_World_Cup_at_2021-22_St._Moritz%E2%80%93Celerina_Luge_World_Cup_and_European_Championships_by_Sandro_Halank%E2%80%93257.jpg/288px-2022-01-22_Men%27s_World_Cup_at_2021-22_St._Moritz%E2%80%93Celerina_Luge_World_Cup_and_European_Championships_by_Sandro_Halank%E2%80%93257.jpg",
|
| 43 |
-
"https://upload.wikimedia.org/wikipedia/commons/thumb/9/99/Wiesen_Pippau_%28Crepis_biennis%29-20220624-RM-123950.jpg/224px-Wiesen_Pippau_%28Crepis_biennis%29-20220624-RM-123950.jpg",
|
| 44 |
-
]
|
| 45 |
-
|
| 46 |
-
#Image.open(requests.get(pic, stream=True).raw).convert("RGB")
|
| 47 |
-
loader = ImageCaptionLoader(path_images=list_image_urls)
|
| 48 |
-
list_docs = loader.load()
|
| 49 |
-
index = VectorstoreIndexCreator().from_loaders([loader])
|
| 50 |
|
| 51 |
button = st.button("Generate Caption")
|
| 52 |
-
query = st.text_area("Enter your query 🔍")
|
| 53 |
|
| 54 |
if button:
|
| 55 |
-
|
| 56 |
-
# Get Response
|
| 57 |
-
caption = index.query(query)
|
| 58 |
-
st.write(caption)
|
|
|
|
| 1 |
import torch
|
| 2 |
import re
|
| 3 |
+
import langchain
|
| 4 |
from PIL import Image
|
| 5 |
import requests
|
| 6 |
import streamlit as st
|
|
|
|
| 32 |
}
|
| 33 |
</style>""", unsafe_allow_html=True)
|
| 34 |
|
| 35 |
+
image = st.file_uploader("Upload an image", type="jpg")
|
| 36 |
+
caption = langchain.generate_caption(image)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
button = st.button("Generate Caption")
|
|
|
|
| 39 |
|
| 40 |
if button:
|
| 41 |
+
st.write("Caption:", caption)
|
|
|
|
|
|
|
|
|