Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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([
|