Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import tensorflow as tf
 from tensorflow.keras.models import load_model
+from transformers import pipeline
+
 import os
 import cv2
 import numpy as np
@@ -126,4 +128,24 @@ with iforest:
 with gan:
     st.write('Próximamente')
 with vit:
-    st.write('Próximamente')
+
+    col_a, col_b = st.columns(2)
+
+    with col_a:
+        st.title('Visual Transformer')
+        st.caption('One class is all you need!')
+        uploaded_file = st.file_uploader(label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
+        if st.button('¿Hay un patacón en la imagen?'):
+            if uploaded_file is not None:
+                classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
+                img = preprocess(uploaded_file)
+                raw_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+                img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))     
+                classifier(img)
+            else:
+                st.write("Asegúrate de haber subido correctamente la imagen.")
+                
+            
+        
+        
+