lihaoxin2020 commited on
Commit
d2b4083
·
1 Parent(s): 56daf6d
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -608,6 +608,10 @@ def synthandreturn(text):
608
  }
609
  # result = router.predict(text, AVAILABLE_MODELS[model].lower(), api_name="/synthesize")
610
  router = Client(model_links[model])
 
 
 
 
611
  result = router.predict(model_args[model], **model_kwargs[model])
612
  else:
613
  # result = router.predict(text, model.lower(), api_name="/synthesize")
@@ -630,6 +634,8 @@ def synthandreturn(text):
630
  except:
631
  raise gr.Error('Unable to call API, please try again :)')
632
  print('Done with', model)
 
 
633
  # try:
634
  # doresample(result)
635
  # except:
 
608
  }
609
  # result = router.predict(text, AVAILABLE_MODELS[model].lower(), api_name="/synthesize")
610
  router = Client(model_links[model])
611
+ # debug
612
+ print(model_args[model])
613
+ print(model_kwargs[model])
614
+
615
  result = router.predict(model_args[model], **model_kwargs[model])
616
  else:
617
  # result = router.predict(text, model.lower(), api_name="/synthesize")
 
634
  except:
635
  raise gr.Error('Unable to call API, please try again :)')
636
  print('Done with', model)
637
+ if model in {"moe"}:
638
+ result = result[1]
639
  # try:
640
  # doresample(result)
641
  # except: