Spaces:
Sleeping
Sleeping
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: | |
imgclassifier(image) | |
return description | |
demo = gr.Interface( | |
fn=guessanImage, | |
inputs=[gr.Image()], | |
outputs=["text"], | |
) | |
demo.launch() | |