Spaces:

gowtham58
/

TNActors_Classifier

Sleeping

gowtham58 commited on Mar 5, 2024

Commit

a585d20

•

1 Parent(s): 2b069b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,9 +4,8 @@ plt = platform.system()
 if plt == 'Linux':
     pathlib.WindowsPath = pathlib.PosixPath
 from fastai.vision.all import *
-import torch
 import gradio as gr
-learn = torch.load('export.pkl',  map_location=torch.device('cpu'))
 actors = learn.dls.vocab
 def classify_images(img):

 if plt == 'Linux':
     pathlib.WindowsPath = pathlib.PosixPath
 from fastai.vision.all import *
 import gradio as gr
+learn = load_learner(Path('export.pkl'))
 actors = learn.dls.vocab
 def classify_images(img):