Spaces:

Ashrafb
/

VT45

Paused

Ashrafb commited on Nov 11, 2024

Commit

d081ae5

verified ·

1 Parent(s): 2a9c47a

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -19,7 +19,7 @@ detector = MTCNN()
 def load_model():
     global model
     model = Model(device='cuda' if torch.cuda.is_available() else 'cpu')
-    model.load_model('cartoon4')
 # Configure logging
 logging.basicConfig(level=logging.INFO)

 def load_model():
     global model
     model = Model(device='cuda' if torch.cuda.is_available() else 'cpu')
+    model.load_model('cartoon1')
 # Configure logging
 logging.basicConfig(level=logging.INFO)