Spaces:
Sleeping
Sleeping
mohamedsaeed823
commited on
Commit
•
a7c81c7
1
Parent(s):
69070c7
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
from fastai.vision.all import *
|
2 |
import gradio as gr
|
3 |
|
|
|
|
|
4 |
def is_metal_album(x): return x[0].isupper()
|
5 |
|
6 |
learn = load_learner('model.pkl')
|
@@ -16,4 +18,4 @@ def predict(img):
|
|
16 |
|
17 |
examples = ['metalkinggizz.png', 'popkinggizz.jpeg', 'TI.png','TI2.jpeg']
|
18 |
|
19 |
-
gr.Interface(fn=predict, inputs=gr.Image(height=512,width=512), outputs=gr.Label(num_top_classes=2),examples=examples).launch(share=True)
|
|
|
1 |
from fastai.vision.all import *
|
2 |
import gradio as gr
|
3 |
|
4 |
+
description = 'this is a model that predicts whether an album is metal or pop based on its cover art'
|
5 |
+
|
6 |
def is_metal_album(x): return x[0].isupper()
|
7 |
|
8 |
learn = load_learner('model.pkl')
|
|
|
18 |
|
19 |
examples = ['metalkinggizz.png', 'popkinggizz.jpeg', 'TI.png','TI2.jpeg']
|
20 |
|
21 |
+
gr.Interface(fn=predict, inputs=gr.Image(height=512,width=512), outputs=gr.Label(num_top_classes=2), examples=examples, description=description).launch(share=True)
|