Ayamohamed commited on
Commit
637c47d
·
verified ·
1 Parent(s): e191128

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from huggingface_hub import hf_hub_download
6
  import gradio as gr
7
 
8
  model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="dia_none_classifier_full.pth")
9
- model = torch.load(model_path, map_location=torch.device("cpu"), weights_only=False))
10
  model.eval()
11
 
12
  transform = transforms.Compose([
 
6
  import gradio as gr
7
 
8
  model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="dia_none_classifier_full.pth")
9
+ model = torch.load(model_path, map_location=torch.device("cpu"), weights_only=False)
10
  model.eval()
11
 
12
  transform = transforms.Compose([