Update app.py

app.py

@@ -25,13 +25,16 @@ def generate_text(input_text):
         num_return_sequences=1,
         eos_token_id=tokenizer.eos_token_id,
     )
+    
+    global cleaned_output_text
 
     output_text = tokenizer.decode(output[0], skip_special_tokens=True)
     print(output_text)
 
     # Remove Prompt Echo from Generated Text
+    
     cleaned_output_text = output_text.replace(input_text, "")
-    return cleaned_output_text
+    return  cleaned_output_text
 
 block = gr.Blocks()
 
@@ -39,11 +42,11 @@ block = gr.Blocks()
 with block:
     gr.Markdown("""<h1><center>CogwiseAI falcon7b</center></h1>
     """)
-    chatbot = gr.Chatbot()
+    # chatbot = gr.Chatbot()
     message = gr.Textbox(placeholder='Enter Your Question Here')
     state = gr.State()
     submit = gr.Button("SEND")
-    submit.click(generate_text, inputs=[message, state], outputs=[output_text, state])
+    submit.click(generate_text, inputs=[message, state], outputs=[cleaned_output_text, state])
 
 block.launch(debug = True)