Hugging Face's logo

Spaces:
azaz
/
MamographyClassifier

App Files Community
azaz commited on
Commit
0f0e13d
•
1 Parent(s): 2f9dfd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ from fastai.learner import load_learner
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
  #learn = from_pretrained_fastai("edureyyy/MamographyClassifier/ModelSuperKek.pkl")
6
- learn = load_learner('edureyyy/MamographyClassifier/ModelSuperKek.pkl')
 
7
 
8
  label = ['Cancer', 'No Cancer']
9
 
 
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
  #learn = from_pretrained_fastai("edureyyy/MamographyClassifier/ModelSuperKek.pkl")
6
+ #learn = load_learner('edureyyy/MamographyClassifier/ModelSuperKek.pkl')
7
+ learn = load_learner('ModelSuperKek.pkl')
8
 
9
  label = ['Cancer', 'No Cancer']
10