Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -48,7 +48,7 @@ def multiclass_prediction(classifier, important_class): #made for hf zero-shot p
 API_URL = "https://api-inference.huggingface.co/models"
 headers = {"Authorization": f"Bearer {st.secrets['token']}"}
 
-def query(data, model_name): #HF API
+def query(data, models): #HF API
     response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
     while "error" in response.json(): 
         response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
@@ -204,7 +204,10 @@ with vit:
                     st.write("loop iniciado")
                     for model in models:
                         i+=1
-                        y_gorrito += query(uploaded_file.read(), model)
+                        st.write("y gorrito a cargar")
+                        a =  query(uploaded_file.read(), model)
+                        st.write("query terminado")
+                        y_gorrito += a
                         st.write("y gorrito cargado")
                     y_gorrito /= i
                     st.write("loop terminado")