Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import gradio as gr
+import json
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, pipeline
 
 tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
@@ -14,10 +15,10 @@ nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
 def generate(prompt):
     # generate 10 samples using contrastive search
     outs = nlp(prompt, max_length=76, num_return_sequences=3, do_sample=True, repetition_penalty=1.2, temperature=0.7, top_k=3, early_stopping=True)
-    outs_json=outs.to_json(orient="records")
+    jsonStr = json.dumps(outs)
     print(prompt)
-    print(outs_json)
-    return outs
+    print(jsonStr)
+    return jsonStr
 
 
 # for i in range(len(outs)):