sergioska commited on
Commit
3e52111
·
1 Parent(s): e0bb95f

fix load image

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -37,15 +37,14 @@ if uploaded_image_file is not None:
37
  st.title('Upload an image file to detection')
38
 
39
  uploaded_image_zero_file = st.file_uploader("Choose an image file (zero)")
40
- texts = st.text_input('apple, eggs')
41
-
42
  if uploaded_image_zero_file is not None:
43
  image = Image.open(uploaded_image_zero_file)
44
  outputImage = np.array(image)
45
  st.image(outputImage)
46
-
47
- if st.button('check tags'):
48
- inputs = processor(text=texts, images=image, return_tensors="pt")
49
  outputs = model(**inputs)
50
  target_sizes = torch.Tensor([image.size[::-1]])
51
  results = processor.post_process_object_detection(outputs=outputs, threshold=0.1, target_sizes=target_sizes)
 
37
  st.title('Upload an image file to detection')
38
 
39
  uploaded_image_zero_file = st.file_uploader("Choose an image file (zero)")
40
+ texts = st.text_input()
41
+ image = ''
42
  if uploaded_image_zero_file is not None:
43
  image = Image.open(uploaded_image_zero_file)
44
  outputImage = np.array(image)
45
  st.image(outputImage)
46
+ tags = texts.split(", ")
47
+ inputs = processor(text=tags, images=image, return_tensors="pt")
 
48
  outputs = model(**inputs)
49
  target_sizes = torch.Tensor([image.size[::-1]])
50
  results = processor.post_process_object_detection(outputs=outputs, threshold=0.1, target_sizes=target_sizes)