Update app.py

app.py

@@ -44,4 +44,21 @@ class ModelWrapper:
         generated_text = ""
         for new_text in streamer:
             generated_text += new_text
-            yield generated_text
+            yield generated_text
+
+model_wrapper = ModelWrapper()
+
+interface = gr.Interface(
+    fn=model_wrapper.generate,
+    inputs=gr.Textbox(lines=5, label="Input Prompt"),
+    outputs=gr.Textbox(label="Generated Text"),
+    title="Mistral-Large-Instruct-2407 Text Completion",
+    description="Enter a prompt and receive a text completion using the Mistral-Large-Instruct-2407 INT4 model.",
+    allow_flagging='never',
+    live=False,
+    cache_examples=False,
+    streaming=True
+)
+
+if __name__ == "__main__":
+    interface.launch()