Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import time
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 prompt=''' Review the Following text as a human, who is tasked to Extract `Code Snippets` ,if any. If there are no Code Snippets in the below Text then return No CODE HIDDEN:
 
@@ -16,7 +17,9 @@ def chatbot_model(m,history):
 
     # Decode the output
     decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return decoded_output
+    for i in range(len(decoded_output)):
+        time.sleep(0.3)
+        yield decoded_output[:i+1]
 
-iface = gr.ChatInterface(chatbot_model)
+iface = gr.ChatInterface(chatbot_model,title="Code Detector",undo_btn=None,theme="soft").queue()
 iface.launch()