Spaces:
Runtime error
Runtime error
visualize beam in output
Browse files
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 |
-
|
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
|