Aruno commited on
Commit
821d811
1 Parent(s): 033440a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,7 +7,7 @@ image_processor = AutoImageProcessor.from_pretrained("Aruno/gemini-beauty")
7
  model = MobileNetV2Model.from_pretrained("Aruno/gemini-beauty")
8
 
9
  def inference(img):
10
- inputs = image_processor(image, return_tensors="pt")
11
  with torch.no_grad():
12
  outputs = model(**inputs)
13
  return outputs
@@ -16,8 +16,8 @@ iface = gradio.Interface(
16
  fn=inference,
17
  inputs='image',
18
  outputs='image',
19
- title='Hello World',
20
- description='The simplest interface!',
21
- examples=[])
22
 
23
  iface.launch()
 
7
  model = MobileNetV2Model.from_pretrained("Aruno/gemini-beauty")
8
 
9
  def inference(img):
10
+ inputs = image_processor(img, return_tensors="pt")
11
  with torch.no_grad():
12
  outputs = model(**inputs)
13
  return outputs
 
16
  fn=inference,
17
  inputs='image',
18
  outputs='image',
19
+ title='Your Attractivness',
20
+ description='Check your attractivness',
21
+ examples=["face_01.jpg", "face_02.jpg"])
22
 
23
  iface.launch()