Amanda Sarubbi commited on
Commit
2177bfe
·
1 Parent(s): 1be441e
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,7 +7,6 @@ from fastai.vision.all import *
7
  import gradio as gr
8
  #import pathlib
9
  from huggingface_hub import hf_hub_download
10
- from huggingface_hub import from_pretrained_fastai
11
  #########################################################################
12
  # user access token for HF model library
13
  ACCESS_TOKEN = "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"
@@ -23,7 +22,7 @@ def import_model(model_name):
23
  use_auth_token=ACCESS_TOKEN,
24
  repo_type='model')
25
 
26
- learn = from_pretrained_fastai(path, cpu=True)
27
 
28
  return learn
29
  #########################################################################
 
7
  import gradio as gr
8
  #import pathlib
9
  from huggingface_hub import hf_hub_download
 
10
  #########################################################################
11
  # user access token for HF model library
12
  ACCESS_TOKEN = "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"
 
22
  use_auth_token=ACCESS_TOKEN,
23
  repo_type='model')
24
 
25
+ learn = load_learner(path, cpu=True)
26
 
27
  return learn
28
  #########################################################################