Spaces:

GerryRaz
/

LungClassifier

Sleeping

GerryRaz commited on Feb 13

Commit

48788e5

verified ·

1 Parent(s): 4641fa2

Bug Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ densenet121, transforms = model.create_densenet_model(num_classes=2)
 state_dict = torch.load(
-    f="/content/dense_90_tiny_lung_classifier_model.pth",
     map_location="cpu",
     weights_only=False
 )
@@ -39,7 +39,7 @@ def predict(img):
           return pred_labels_and_probs, pred_time
-example_list = '/content/examples' # The path to your directory
 import gradio as gr

 state_dict = torch.load(
+    f="dense_90_tiny_lung_classifier_model.pth",
     map_location="cpu",
     weights_only=False
 )
           return pred_labels_and_probs, pred_time
+example_list = 'examples' # The path to your directory
 import gradio as gr