import gradio
from transformers import AutoImageProcessor, MobileNetV2Model
import torch
from datasets import load_dataset

image_processor = AutoImageProcessor.from_pretrained("Aruno/gemini-beauty")
model = MobileNetV2Model.from_pretrained("Aruno/gemini-beauty")

def inference(img):
    inputs = image_processor(image, return_tensors="pt")
    with torch.no_grad():
        outputs = model(**inputs)
    return outputs

iface = gradio.Interface(
  fn=inference,
  inputs='image',
  outputs='image',
  title='Hello World', 
  description='The simplest interface!',
  examples=[])  

iface.launch()