json

app.py

@@ -17,7 +17,6 @@ def extract_json(gen_text, n_shot_learning=0):
         for i in range(0, n_shot_learning):
             gen_text = gen_text[start_index:]
             start_index = gen_text.index("### Response:\n{") + 14
-
     end_index = gen_text.index("}\n\n### ") + 1
     return gen_text[start_index:end_index]
 
@@ -47,9 +46,8 @@ def get_completion(prompt):
 def greet(input):
 	response = get_completion(input)
 	gen_text = response["predictions"][0]["generated_text"]
-
 	return extract_json(gen_text, 3)
 	#return json.dumps(response)
 
-iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Prompt", lines=3)], outputs="text")
+iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Prompt", lines=3)], outputs="json")
 iface.launch()