Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(),
|