phind commited on
Commit
e223dbc
·
1 Parent(s): 1b66fd4
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -3,12 +3,12 @@ from transformers import pipeline
3
  from PIL import Image
4
 
5
  @st.cache_resource
6
- def get_model_hotdog_classification_pipeline():
7
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
8
- return pipeline
9
 
10
  @st.cache_resource
11
- def get_model_image_captioning_pipeline():
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
- pipeline = get_model_hotdog_classification_pipeline()
27
- predictions = pipeline(image)
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 = get_model_image_captioning_pipeline()
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: