Update app.py

app.py

@@ -23,8 +23,9 @@ def generate(query, temp, top_p):
         do_sample=True,
         temperature=temp,
         top_p=top_p)
-    outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
+    outputs = tokenizer.decode(outputs[0][input_length:], skip_special_tokens=True)    
     return outputs
+    
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     input = gr.Textbox(label="Prompt", value="Pripun kulo saged nyinaoni Basa Jawa kanthi sae?")
     output = gr.Textbox(label="Response", scale=2)