Spaces:

pvyas96
/

thin_section_prediction

Runtime error

pvyas96 commited on Apr 10

Commit

0eb9c2a

•

1 Parent(s): 6903545

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from fastai.vision.all import *
 import gradio as gr
-learn = load_learner('model.pth')
 categories = ('Andalusite', 'Argillaceous_siltstone', 'Bioturbated_siltstone', 'Massive_calcareous_siltstone', 'Massive_calcite-cemented_siltstone', 'Porous_calcareous_siltstone', 'actinolite', 'biotite', 'carbonate', 'coal', 'debris_rock', 'granite', 'hornblende', 'olivine', 'oolites', 'plagioclase', 'pyroxene', 'sandstone', 'staurolite ')
 def classify_image(img):

 import gradio as gr
+learn = load_learner('petronet_101.pth')
 categories = ('Andalusite', 'Argillaceous_siltstone', 'Bioturbated_siltstone', 'Massive_calcareous_siltstone', 'Massive_calcite-cemented_siltstone', 'Porous_calcareous_siltstone', 'actinolite', 'biotite', 'carbonate', 'coal', 'debris_rock', 'granite', 'hornblende', 'olivine', 'oolites', 'plagioclase', 'pyroxene', 'sandstone', 'staurolite ')
 def classify_image(img):