Spaces:
Running
Running
File size: 632 Bytes
25b5a23 821d811 25b5a23 821d811 25b5a23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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(img, return_tensors="pt")
with torch.no_grad():
outputs = model(**inputs)
return outputs
iface = gradio.Interface(
fn=inference,
inputs='image',
outputs='image',
title='Your Attractivness',
description='Check your attractivness',
examples=["face_01.jpg", "face_02.jpg"])
iface.launch() |