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 = "microsoft/resnet-50"
    # st.write("using model:"+model)
    
    imgclassifier  = pipeline("image-classification", model=model)
    if image is not None:  
        b64img = base64.b64encode(image.getvalue()).decode()
        imgclassifier(b64img)
        
    return description

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

demo.launch()