Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import tensorflow as tf
 from tensorflow.keras.models import load_model
-from transformers import AutoConfig, AutoModel, pipeline
+from transformers import AutoConfig, AutoModel, pipeline, AutoProcessor, AutoModelForZeroShotImageClassification
 from PIL import Image 
 import os
 import cv2
@@ -50,13 +50,15 @@ headers = {"Authorization": f"Bearer {st.secrets['token']}"}
 
 def query(data, model_name): #HF API
     response = requests.post(API_URL + "/" + model_name, headers=headers, data=data)
-    with st.spinner("Cargando Predicción... (si se demora un poco, es probable que esté cargando el modelo)"):
-        while "error" in response.json(): 
-            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)
+    st.write("Done.")
     return response.json() #.json
 
 @st.cache_resource
 def load_clip():
+    #processor = AutoProcessor.from_pretrained("openai/clip-vit-large-patch14-336")
+    #model = AutoModelForZeroShotImageClassification.from_pretrained("openai/clip-vit-large-patch14-336")
     classifier = pipeline("zero-shot-image-classification", model = 'openai/clip-vit-large-patch14-336')
     return classifier