Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -50,7 +50,8 @@ headers = {"Authorization": f"Bearer {st.secrets['token']}"}
 
 def query(data, models): #HF API
     response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
-    st.write(response.json())
+    if response.json()["error"] == "Internal Server Error":
+        return -1
     while "error" in response.json(): 
         response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
     return response.json()[1]["score"] #.json
@@ -176,15 +177,15 @@ with vit:
             elif uploaded_file is not None:
                 with st.spinner('Cargando predicción...'):
 
-                    y_gorritoo = query(uploaded_file.read(), model_dict[model_choice[0]]) 
-                    st.write(y_gorritoo)
+                    #y_gorritoo = query(uploaded_file.read(), model_dict[model_choice[0]]) 
+                    #st.write(y_gorritoo)
                     #classifiers = [pipeline("image-classification", model= model_dict[model_choice[i]]) for i in range(len(model_choice))]
                     
                     #classifier = pipeline("image-classification", model= model_dict[model_choice[0]])
                     img = preprocess(uploaded_file, module = 'pil')
 
                     models = [model_dict[model] for model in model_choice]
-                    st.write(models)
+                    #st.write(models)
                     #def vit_ensemble(classifier_list, img):
                     #   y_gorrito = 0
                     #    for classifier in classifier_list:
@@ -207,6 +208,9 @@ with vit:
                         i+=1
                         st.write("y gorrito a cargar")
                         a =  query(uploaded_file.read(), model)
+                        if a == -1:
+                            st.write("Los servidores se encuentrar caídos, intente más tarde")
+                            break
                         st.write("query terminado")
                         y_gorrito += a
                         st.write("y gorrito cargado")