Update app.py

app.py

@@ -36,10 +36,10 @@ def query(payload):
     print(payload)
     response = requests.request("POST", API_URL, json=payload)
     print(response)
-    return json.loads(response.content.decode("utf-8"))
+    return response.json()
     
 def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
-    payload = {"prompts": input_sentence, "gen_method": sample_or_greedy, "max_new_tokens": max_length}
+    payload = {"prompt": input_sentence, "gen_method": sample_or_greedy, "max_new_tokens": max_length}
 
     data = query(
         payload
@@ -64,7 +64,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
         BOX_COMP: "#120F25",
     }
     
-    new_string = data[0]['generated_text'].split(input_sentence, 1)[1]
+    new_string = data['generated_text'][0].split(input_sentence, 1)[1]
 
     _, img = main(
         input_sentence,
@@ -79,7 +79,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
         init_font_size=150,
         right_align=False,
     )
-    return img, data[0]['generated_text']
+    return img, data['generated_text'][0]
 
 
 gr.Interface(