visualize beam in output

app.py

@@ -2,11 +2,20 @@ import os
 import sys
 import requests
 import gradio as gr 
+import json
+import numpy as np
+from io import BytesIO
+from PIL import Image
 
 def translate(input_txt,prefix,suffix,num_beams,topk,beta):
     dictToSend = {'input':input_txt,'prefix':prefix,'suffix':suffix,'num_beams':num_beams,'alpha': (1-beta),'beta':beta,'topk':topk}
-    res = requests.post('https://d7bf-141-3-25-29.ngrok-free.app/generate', json=dictToSend)
-    return res.text,res.beam_img
+    response = requests.post('https://d7bf-141-3-25-29.ngrok-free.app/generate', json=dictToSend)
+    response_data = json.loads(response.text)
+    processed_text = response_data.get('text')
+    processed_img = response_data.get('beam_img')
+    processed_image_array = np.array(processed_image_list, dtype=np.uint8)
+    processed_image = Image.fromarray(processed_image_array)
+    return processed_text, processed_image
 
 prefix= "<s>[INST] <<SYS>>\nYou are a professional translator from English to German. You translate the sentences to only Formal German even if the English sentence is Informal.\n<</SYS>>\nEnglish: "
 #self.prefix = "Translate from English to German:\nEnglish: "

requirements.txt

@@ -0,0 +1,3 @@
+requests
+numpy
+Pillow