Spaces:
Sleeping
Sleeping
phind
commited on
Commit
·
e223dbc
1
Parent(s):
1b66fd4
fix bugs
Browse files
app.py
CHANGED
@@ -3,12 +3,12 @@ from transformers import pipeline
|
|
3 |
from PIL import Image
|
4 |
|
5 |
@st.cache_resource
|
6 |
-
def
|
7 |
-
|
8 |
-
return
|
9 |
|
10 |
@st.cache_resource
|
11 |
-
def
|
12 |
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
|
13 |
return captioner
|
14 |
|
@@ -23,8 +23,8 @@ with tabs1:
|
|
23 |
|
24 |
image = Image.open(file_name)
|
25 |
col1.image(image, use_column_width=True)
|
26 |
-
|
27 |
-
predictions =
|
28 |
|
29 |
col2.header("Probabilities")
|
30 |
for p in predictions:
|
@@ -38,7 +38,7 @@ with tabs2:
|
|
38 |
if file_name is not None:
|
39 |
image = Image.open(file_name)
|
40 |
st.image(image, use_column_width=True)
|
41 |
-
captioner =
|
42 |
predictions = captioner(image)
|
43 |
st.write("**generated_text: **")
|
44 |
for generated_text in predictions:
|
|
|
3 |
from PIL import Image
|
4 |
|
5 |
@st.cache_resource
|
6 |
+
def get_model_hotdog_classification():
|
7 |
+
model = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
|
8 |
+
return model
|
9 |
|
10 |
@st.cache_resource
|
11 |
+
def get_model_image_captioning():
|
12 |
captioner = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
|
13 |
return captioner
|
14 |
|
|
|
23 |
|
24 |
image = Image.open(file_name)
|
25 |
col1.image(image, use_column_width=True)
|
26 |
+
model = get_model_hotdog_classification()
|
27 |
+
predictions = model(image)
|
28 |
|
29 |
col2.header("Probabilities")
|
30 |
for p in predictions:
|
|
|
38 |
if file_name is not None:
|
39 |
image = Image.open(file_name)
|
40 |
st.image(image, use_column_width=True)
|
41 |
+
captioner = get_model_image_captioning()
|
42 |
predictions = captioner(image)
|
43 |
st.write("**generated_text: **")
|
44 |
for generated_text in predictions:
|