mrfakename commited on
Commit
ac01906
1 Parent(s): 4bd9385

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -606,14 +606,21 @@ def synthandreturn(text):
606
  pass
607
  if model in AVAILABLE_MODELS.keys(): model = AVAILABLE_MODELS[model]
608
  print(model)
 
609
  result_storage[model] = result
610
  # try:
611
  # doloudnorm(result)
612
  # except:
613
  # pass
 
 
 
 
 
614
  results = {}
615
- thread1 = threading.Thread(target=predict_and_update_result, args=(text, mdl1, results))
616
- thread2 = threading.Thread(target=predict_and_update_result, args=(text, mdl2, results))
 
617
 
618
  thread1.start()
619
  thread2.start()
@@ -629,12 +636,8 @@ def synthandreturn(text):
629
  # print(sr)
630
  #debug
631
  # outputs = [text, btn, r2, model1, model2, aud1, aud2, abetter, bbetter, prevmodel1, prevmodel2, nxtroundbtn]
632
- mdl1k = mdl1
633
- mdl2k = mdl2
634
- print(mdl1k, mdl2k)
635
- if mdl1 in AVAILABLE_MODELS.keys(): mdl1k=AVAILABLE_MODELS[mdl1]
636
- if mdl2 in AVAILABLE_MODELS.keys(): mdl2k=AVAILABLE_MODELS[mdl2]
637
 
 
638
  return (
639
  text,
640
  "Synthesize",
 
606
  pass
607
  if model in AVAILABLE_MODELS.keys(): model = AVAILABLE_MODELS[model]
608
  print(model)
609
+ print(f"Running model {model}")
610
  result_storage[model] = result
611
  # try:
612
  # doloudnorm(result)
613
  # except:
614
  # pass
615
+ mdl1k = mdl1
616
+ mdl2k = mdl2
617
+ print(mdl1k, mdl2k)
618
+ if mdl1 in AVAILABLE_MODELS.keys(): mdl1k=AVAILABLE_MODELS[mdl1]
619
+ if mdl2 in AVAILABLE_MODELS.keys(): mdl2k=AVAILABLE_MODELS[mdl2]
620
  results = {}
621
+ print(f"Sending models {mdl1} and {mdl2} to API")
622
+ thread1 = threading.Thread(target=predict_and_update_result, args=(text, mdl1k, results))
623
+ thread2 = threading.Thread(target=predict_and_update_result, args=(text, mdl2k, results))
624
 
625
  thread1.start()
626
  thread2.start()
 
636
  # print(sr)
637
  #debug
638
  # outputs = [text, btn, r2, model1, model2, aud1, aud2, abetter, bbetter, prevmodel1, prevmodel2, nxtroundbtn]
 
 
 
 
 
639
 
640
+ print(f"Retrieving models {mdl1} and {mdl2} from API")
641
  return (
642
  text,
643
  "Synthesize",