Hugging Face's logo

Spaces:
Sonogram
/
Instructor-Support-Tool
Sleeping

App Files Community
czyoung commited on
Commit
996b299
·
verified ·
1 Parent(s): 1a1a902

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,9 +48,9 @@ except RuntimeError as e:
48
 
49
  # Instantiate and prepare model for training.
50
  dfModel, dfState, _ = init_df(model_base_dir="DeepFilterNet3")
51
- dfModel.to(torch.device("cuda"))
52
  pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
53
- pipeline.to(torch.device("cuda"))
54
 
55
  # Store results for viewing and further processing
56
  if 'results' not in st.session_state:
 
48
 
49
  # Instantiate and prepare model for training.
50
  dfModel, dfState, _ = init_df(model_base_dir="DeepFilterNet3")
51
+ dfModel.to(device)#torch.device("cuda"))
52
  pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
53
+ pipeline.to(device)#torch.device("cuda"))
54
 
55
  # Store results for viewing and further processing
56
  if 'results' not in st.session_state: