sergioska commited on
Commit
8de96d1
·
1 Parent(s): 42c5bd8

trying to print image with rectagles

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,7 +3,6 @@ import streamlit as st
3
  import numpy as np
4
  from PIL import Image, ImageDraw
5
  from transformers import pipeline
6
- from transformers import OwlViTProcessor, OwlViTForObjectDetection
7
  from tempfile import NamedTemporaryFile
8
 
9
  audiopipe = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
@@ -69,6 +68,6 @@ if st.button('apply tag'):
69
  draw.rectangle((xmin, ymin, xmax, ymax), outline="red", width=1)
70
  draw.text((xmin, ymin), f"{label}: {round(score,2)}", fill="white")
71
 
72
- #st.image(image)
73
  st.write(predictions)
74
 
 
3
  import numpy as np
4
  from PIL import Image, ImageDraw
5
  from transformers import pipeline
 
6
  from tempfile import NamedTemporaryFile
7
 
8
  audiopipe = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
 
68
  draw.rectangle((xmin, ymin, xmax, ymax), outline="red", width=1)
69
  draw.text((xmin, ymin), f"{label}: {round(score,2)}", fill="white")
70
 
71
+ st.image(image)
72
  st.write(predictions)
73