Update app.py

app.py

@@ -109,7 +109,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False, seed
     data = query(payload)
 
     if raw_text:
-        return None, data[0]["generated_text"]
+        return None, data["generated_text"]
 
     width, height = 3246, 3246
     assets_path = "assets"
@@ -127,9 +127,9 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False, seed
         BOX_COMP: "#120F25",
     }
 
-    new_string = data[0]["generated_text"].split(input_sentence, 1)[1]
+    new_string = data["generated_text"].split(input_sentence, 1)[1]
 
-    return data[0]["generated_text"]
+    return data["generated_text"]
 
 
 gr.Interface(