Hugging Face's logo

Spaces:
Woziii
/
scribe
Running

App Files Community
Woziii commited on
Commit
71a214b
·
verified ·
1 Parent(s): dc53ac1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -45,6 +45,7 @@ pipe = pipeline(
45
  device=device,
46
  model_kwargs={"low_cpu_mem_usage": True},
47
  )
 
48
 
49
 
50
 
 
45
  device=device,
46
  model_kwargs={"low_cpu_mem_usage": True},
47
  )
48
+ pipe.to('cuda')
49
 
50
 
51