Mohaddz commited on
Commit
cd9bb6e
·
verified ·
1 Parent(s): 407491f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ class MultiClientThemeClassifier:
36
  self.model_loaded = False
37
 
38
  print(f"Loading model: {model_name} onto CUDA device")
39
- self.model = SentenceTransformer(model_name, device='cuda', trust_remote_code=True,token_hf)
40
  self.model_loaded = True
41
  self.current_model_name = model_name
42
  return f"✅ Model '{model_name}' loaded successfully onto GPU!"
 
36
  self.model_loaded = False
37
 
38
  print(f"Loading model: {model_name} onto CUDA device")
39
+ self.model = SentenceTransformer(model_name, device='cuda', trust_remote_code=True,token=token_hf)
40
  self.model_loaded = True
41
  self.current_model_name = model_name
42
  return f"✅ Model '{model_name}' loaded successfully onto GPU!"