Update app.py

app.py

@@ -52,7 +52,7 @@ class VideoClassifier:
     def setup_mistral_model(self):
         self.model_id = "mistralai/Mistral-7B-Instruct-v0.2"
         self.device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
-        self.device_name = torch.cuda.get_device_name()
+        # self.device_name = torch.cuda.get_device_name()
         # print(f"Using device: {self.device} ({self.device_name})")
         bnb_config = transformers.BitsAndBytesConfig(
             load_in_4bit=True,