sync ms

README.md

@@ -4,7 +4,7 @@ emoji: ☰
 colorFrom: red
 colorTo: indigo
 sdk: gradio
-sdk_version: 6.3.0
+sdk_version: 6.6.0
 app_file: app.py
 pinned: true
 license: mit

model.py

@@ -21,10 +21,10 @@ class EvalNet:
         self.model = eval("models.%s()" % m_ver)
         linear_output = self._set_outsize()
         self._set_classifier(cls_num, linear_output)
-        checkpoint = torch.load(saved_model_path, map_location="cpu")
-        if torch.cuda.is_available():
-            checkpoint = torch.load(saved_model_path)
-
+        checkpoint = torch.load(
+            saved_model_path,
+            map_location=torch.device("cuda:0") if torch.cuda.is_available() else "cpu",
+        )
         self.model.load_state_dict(checkpoint, False)
         self.model.eval()