IA1 / IA.py
sorizano's picture
Create IA.py
f979f15
raw
history blame contribute delete
898 Bytes
import streamlit as st
from PIL import Image
from transformers import pipeline
from google.colab import drive
drive.mount('/content/drive')
ruta_google_drive = "/content/drive/MyDrive/Lenguaje/"
translator = pipeline("translation", model="google/mbt-youtube-100k")
letras = ['A', 'B', 'C']
categorias = ['Frutas', 'Colores', 'Familiares', 'Animales']
st.title("Proyecto de Lenguaje de Señas")
opcion = st.sidebar.selectbox("Seleccionar una opción", ["Letras", "Categorías"])
if opcion == "Letras":
letra = st.selectbox("Seleccionar una letra", letras)
st.write(f"Lenguaje de señas para la letra {letra}")
st.image(f"ruta_a_tu_google_drive/A.jpg")
elif opcion == "Categorías":
categoria = st.selectbox("Seleccionar una categoría", categorias)
st.write(f"Lenguaje de señas para la categoría {categoria}")
st.image(f"ruta_a_tu_google_drive/{categoria}.jpg")