miwojc commited on
Commit
8ebcaf2
·
1 Parent(s): 520d98a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from fastai.vision.all import *
3
  from huggingface_hub import from_pretrained_fastai
4
 
5
- repo_id = "hugginglearners/brain-tumor-detection-mri"
6
- learn = from_pretrained_fastai(repo_id)
7
  labels = learn.dls.vocab
8
 
9
  def predict(img):
 
2
  from fastai.vision.all import *
3
  from huggingface_hub import from_pretrained_fastai
4
 
5
+ #repo_id = "hugginglearners/brain-tumor-detection-mri"
6
+ learn = from_pretrained_fastai("hugginglearners/brain-tumor-detection-mri")
7
  labels = learn.dls.vocab
8
 
9
  def predict(img):