frncscp commited on
Commit
75736a2
·
1 Parent(s): ff640ab

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +1 -3
pages/Entorno de Ejecución.py CHANGED
@@ -144,15 +144,13 @@ with vit:
144
  with st.spinner('Cargando predicción...'):
145
  classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
146
  img = preprocess(uploaded_file, module = 'pil')
147
- raw_img = img
148
- img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))
149
  flag = True
150
  else:
151
  st.write("Asegúrate de haber subido correctamente la imagen.")
152
  with col_b:
153
  if flag:
154
  st.write(classifier(img))
155
- st.image(raw_img)
156
 
157
  with zero_shot:
158
  st.write('Próximamente')
 
144
  with st.spinner('Cargando predicción...'):
145
  classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
146
  img = preprocess(uploaded_file, module = 'pil')
 
 
147
  flag = True
148
  else:
149
  st.write("Asegúrate de haber subido correctamente la imagen.")
150
  with col_b:
151
  if flag:
152
  st.write(classifier(img))
153
+ st.image(img)
154
 
155
  with zero_shot:
156
  st.write('Próximamente')