Update app.py

app.py

@@ -1,4 +1,5 @@
 import torch
+import gradio as gr
 
 from fastapi import FastAPI
 from pydantic import BaseModel
@@ -14,4 +15,10 @@ class InferenceRequest(BaseModel):
 def predict(request: InferenceRequest):
     return {
         'generated_text': request.prompt
-    }
+    }
+
+# Create the Gradio interface
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+
+# Launch the interface
+iface.launch()