Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 import torch
 
 # Set max_split_size_mb
-os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:50'
+# os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:50'
 title = """# 🙋🏻‍♂️Welcome to🌟Tonic's Nexus🐦‍⬛Raven"""
 description = """You can build with this endpoint using Nexus Raven. The demo is still a work in progress but we hope to add some endpoints for commonly used functions such as intention mappers and audiobook processing.
         You can also use Nexus🐦‍⬛Raven on your laptop & by cloning this space. 🧬🔬🔍 Simply click here: <a style="display:inline-block" href="https://huggingface.co/spaces/Tonic1/NexusRaven2?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></h3> 
@@ -37,12 +37,18 @@ def create_interface():
         submit_button.click(converter.process_text, inputs=input_text, outputs=output_text)
     return app
 
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown(title)
+        gr.Markdown(description)
+        with gr.Row():
+            input_text = gr.Textbox(label="Input Text", placeholder="Enter your text here...")
+            submit_button = gr.Button("Submit")
+        output_text = gr.Textbox(label="Nexus🐦‍⬛Raven", placeholder="Generated text will appear here...")
+        submit_button.click(process_text, inputs=input_text, outputs=output_text)
+
+    demo.launch()
+
 if __name__ == "__main__":
-    demo = gr.Interface(
-        fn=process_text,
-        inputs="text",
-        outputs="text",
-        title=title,
-        description=description
-    )
-    demo.launch()
+    main()