Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -49,11 +49,10 @@ API_URL = "https://api-inference.huggingface.co/models"
 headers = {"Authorization": f"Bearer {st.secrets['token']}"}
 
 def query(data, model_name): #HF API
-    #with open(filename, "rb") as f:
-    #    data = f.read()
-    model_api = API_URL + "/" + model_name
-    st.write(model_api)
-    response = requests.post(model_api, headers=headers, data=data)
+    response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
+    while "error" in response.json():
+        with st.spinner("El modelo está cargando, luego hace predicciones rápidas")
+        response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
     return response.json() #.json
 
 @st.cache_resource