dx2102 commited on
Commit
d3bf393
·
verified ·
1 Parent(s): 3b9ae98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,7 @@ cpu_pipe = transformers.pipeline(
34
  device='cpu',
35
  )
36
  # print devices
37
- print(gpu_pipe.device, cpu_pipe.device)
38
  print('Done')
39
 
40
  example_prefix = '''pitch duration wait velocity instrument
@@ -199,6 +199,7 @@ CPUs will be slower but there is no time limit.
199
  def end(self):
200
  queue.put(None)
201
  def background_fn():
 
202
  result = pipe(
203
  prefix,
204
  streamer=MyStreamer(),
 
34
  device='cpu',
35
  )
36
  # print devices
37
+ print(f"{gpu_pipe.device = }, {cpu_pipe.device = }")
38
  print('Done')
39
 
40
  example_prefix = '''pitch duration wait velocity instrument
 
199
  def end(self):
200
  queue.put(None)
201
  def background_fn():
202
+ print(f"{pipe.device = })")
203
  result = pipe(
204
  prefix,
205
  streamer=MyStreamer(),