frncscp commited on
Commit
25ba27a
1 Parent(s): 1ba6a43

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +3 -1
pages/Entorno de Ejecución.py CHANGED
@@ -173,6 +173,8 @@ with vit:
173
  elif uploaded_file is not None:
174
  with st.spinner('Cargando predicción...'):
175
 
 
 
176
  classifiers = [pipeline("image-classification", model= model_dict[model_choice[i]]) for i in range(len(model_choice))]
177
 
178
  #classifier = pipeline("image-classification", model= model_dict[model_choice[0]])
@@ -195,7 +197,7 @@ with vit:
195
  # y_gorrito = clase["score"]
196
 
197
  #y_gorrito = classifier[0]["score"]
198
- api_result = query(img, model_choice[0])
199
  st.write(api_result)
200
  y_gorrito = vit_ensemble(classifiers, img)
201
 
 
173
  elif uploaded_file is not None:
174
  with st.spinner('Cargando predicción...'):
175
 
176
+ api_result = query(uploaded_file.read(), model_choice[0])
177
+
178
  classifiers = [pipeline("image-classification", model= model_dict[model_choice[i]]) for i in range(len(model_choice))]
179
 
180
  #classifier = pipeline("image-classification", model= model_dict[model_choice[0]])
 
197
  # y_gorrito = clase["score"]
198
 
199
  #y_gorrito = classifier[0]["score"]
200
+
201
  st.write(api_result)
202
  y_gorrito = vit_ensemble(classifiers, img)
203