Update app.py

app.py

@@ -1,30 +1,35 @@
+import gradio as gr
 from transformers import pipeline
 
-pipeline = pipeline(
+raven_pipeline = pipeline(
     "text-generation",
     model="Nexusflow/NexusRaven-V2-13B",
     torch_dtype="auto",
     device_map="auto",
 )
 
-title="""# 🙋🏻‍♂️Welcome to🌟Tonic's🐦‍⬛NexusRaven"""
-description="""this model is used to select and return function calling arguments.
-"""
+class DialogueToSpeechConverter:
+    def __init__(self):
+        self.raven_pipeline = raven_pipeline
 
-prompt = prompt_template.format(query="What's the weather like in Seattle right now?")
+    def process_text(self, input_text: str) -> str:
+        prompt = f"User Query: {input_text}<human_end>"
+        result = self.raven_pipeline(prompt, max_new_tokens=2048, return_full_text=False, do_sample=False, temperature=0.001)[0]["generated_text"]
+        return result
 
-result = pipeline(prompt, max_new_tokens=2048, return_full_text=False, do_sample=False, temperature=0.001)[0]["generated_text"]
-print (result)
+# Gradio interface
+def create_interface():
+    converter = DialogueToSpeechConverter()
+    with gr.Blocks() as app:
+        gr.Markdown("""# 🙋🏻‍♂️Welcome to🌟Tonic's Nexus🐦‍⬛Raven""")
+        gr.Markdown("""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.""")
+        with gr.Row():
+            input_text = gr.Textbox(label="Input Text")
+            output_text = gr.Textbox(label="Nexus🐦‍⬛Raven", readonly=True)
+        input_text.change(converter.process_text, inputs=input_text, outputs=output_text)
 
-with gr.Blocks() as app:
-    gr.Markdown(title)
-    gr.Markdown(description)
-    with gr.Row():
-        with gr.Column():
-            input = gr.Textbox()
-        with gr.Column():
-            output = gr.Textbox()
-    image_input.change(process_image, inputs=image_input, outputs=output)
+    return app
 
 if __name__ == "__main__":
+    app = create_interface()
     app.launch()