YinuoGuo27 commited on
Commit
a7d345b
·
verified ·
1 Parent(s): 50df1fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -56,6 +56,8 @@ def main():
56
  device = "cuda"
57
  else:
58
  device = "cpu"
 
 
59
  kd_talker = Inferencer()
60
 
61
  with gr.Blocks(analytics_enabled=False) as interface:
 
56
  device = "cuda"
57
  else:
58
  device = "cpu"
59
+ from torch.multiprocessing import Pool, Process, set_start_method
60
+ set_start_method('spawn')
61
  kd_talker = Inferencer()
62
 
63
  with gr.Blocks(analytics_enabled=False) as interface: