Update app.py

app.py

@@ -19,7 +19,7 @@ def vipe_generate(text, model, tokenizer,device,do_sample,top_k=100, epsilon_cut
     #return only the generated prompts
     pred_caps = tokenizer.batch_decode(generated_ids[:, -(generated_ids.shape[1] - input_ids.shape[1]):], skip_special_tokens=True)
 
-    return pred_caps
+    return pred_caps[0]
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = GPT2LMHeadModel.from_pretrained('fittar/ViPE-M-CTX7')
@@ -40,7 +40,7 @@ examples = [
 
 title = "ViPE: Visualize Pretty-much Everything"
 description = 'ViPE is the first automated model for translating any arbitrary piece of text into a visualizable prompt. It helps any text-to-image model in figurative or non-lexical language visualizations. To learn more about the model, [click here](https://huggingface.co/fittar/ViPE-M-CTX7).<br>'
-txt = gr.Textbox(lines=1, label="Initial Text", placeholder="Arbitrary Input Text")
+txt = gr.Textbox(lines=1, label="Arbitrary Input Text", placeholder="Initial text")
 out = gr.Textbox(lines=4, label="Generated Prompt for Visualizations")
 
 demo = gr.Interface(