Spaces:
Sleeping
Sleeping
ranjith
commited on
Commit
•
729b213
1
Parent(s):
5890f7a
description added
Browse files- app.py +6 -3
- basalt.jpeg +0 -0
- basalt1.jpeg +0 -0
- limestone.jpeg +0 -0
- limestone1.jpg +0 -0
- marble.jpeg +0 -0
- marble1.jpeg +0 -0
- sandstone.jpeg +0 -0
- sandstone4.jpeg +0 -0
app.py
CHANGED
@@ -13,6 +13,9 @@ def predict(img):
|
|
13 |
pred,pred_idx,probs = learn.predict(img)
|
14 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
15 |
|
16 |
-
title = "
|
17 |
-
examples = ["
|
18 |
-
|
|
|
|
|
|
|
|
13 |
pred,pred_idx,probs = learn.predict(img)
|
14 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
15 |
|
16 |
+
title = "<div style='text-align:center'> Rock classifier</div>"
|
17 |
+
examples = ["basalt.jpeg","coal.jpeg","granite.jpeg", "limestone1.jpeg","marble1.jpeg","quartzite.jpeg","sandstone4.jpeg"]
|
18 |
+
description = "Rock classification using convolutional neural networks results may not be accurate since trained on small dataset"
|
19 |
+
article="<p style='text-align: center'><a href='https://www.linkedin.com/in/amrutha-kp/' target='_blank'>Linkedin</a></p>"
|
20 |
+
|
21 |
+
gr.Interface(fn=predict, inputs=gr.Image(), outputs=gr.Label(num_top_classes=7),title = title,examples= examples,article=article,description=description).launch(share=True)
|
basalt.jpeg
ADDED
basalt1.jpeg
DELETED
Binary file (11.9 kB)
|
|
limestone.jpeg
DELETED
Binary file (5.43 kB)
|
|
limestone1.jpg
ADDED
marble.jpeg
DELETED
Binary file (6.58 kB)
|
|
marble1.jpeg
ADDED
sandstone.jpeg
DELETED
Binary file (8.48 kB)
|
|
sandstone4.jpeg
ADDED