import torch
import os
import random
import gradio as gr
from transformers import pipeline
import base64
from datasets import load_dataset
from diffusers import DiffusionPipeline
from huggingface_hub import login
import numpy as np

def guessanImage(image, model):
    #model = "microsoft/resnet-50"
    # st.write("using model:"+model)
    
    imgclassifier  = pipeline("image-classification", model=model)
    if image is not None:  
        description = imgclassifier(image)
    return description

radio = gr.Radio(["microsoft/resnet-50", "google/vit-base-patch16-224", "apple/mobilevit-small"], label="Select a Classifier", info="Image Classifier?")

demo = gr.Interface(
    fn=guessanImage,
    inputs=[gr.Image(type="pil"), radio],
    outputs=["text"],
)

demo.launch()