skoneru commited on
Commit
8ab8728
·
1 Parent(s): 1d2df92

visualize beam in output

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def translate(input_txt,prefix,suffix,num_beams,topk,beta):
12
  response = requests.post('https://d7bf-141-3-25-29.ngrok-free.app/generate', json=dictToSend)
13
  response_data = json.loads(response.text)
14
  processed_text = response_data.get('text')
15
- processed_img = response_data.get('beam_img')
16
  processed_image_array = np.array(processed_image_list, dtype=np.uint8)
17
  processed_image = Image.fromarray(processed_image_array)
18
  return processed_text, processed_image
 
12
  response = requests.post('https://d7bf-141-3-25-29.ngrok-free.app/generate', json=dictToSend)
13
  response_data = json.loads(response.text)
14
  processed_text = response_data.get('text')
15
+ processed_image_list = response_data.get('beam_img')
16
  processed_image_array = np.array(processed_image_list, dtype=np.uint8)
17
  processed_image = Image.fromarray(processed_image_array)
18
  return processed_text, processed_image