Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -6,6 +6,7 @@ from PIL import Image
 import os
 import cv2
 import numpy as np
+import requests
 
 st.set_page_config(
     page_title = 'Patacotrón',
@@ -44,6 +45,15 @@ def multiclass_prediction(classifier, important_class): #made for hf zero-shot p
             class_score = clase['score']
     return (labels[0] if len(labels) == 1 else labels, score, class_score)
 
+API_URL = "https://api-inference.huggingface.co/models"
+headers = {"Authorization": f"Bearer {token}"}
+
+def query(filename, model_name): #HF API
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.post(API_URL + / + model_name', headers=headers, data=data)
+    return response.json #.json
+
 @st.cache_resource
 def load_clip():
     classifier = pipeline("zero-shot-image-classification", model = 'openai/clip-vit-large-patch14-336')
@@ -185,7 +195,7 @@ with vit:
                     #        y_gorrito = clase["score"]
                 
                 #y_gorrito = classifier[0]["score"]
-
+                    api_result = query(img, model_choice[0])    
                     y_gorrito = vit_ensemble(classifiers, img)
                     
                     if round(float(y_gorrito*100)) >= threshold: