Update app.py

app.py

@@ -4,6 +4,8 @@ from flask import Flask, request
 from flask_caching import Cache
 from flask_cors import CORS
 import codegen
+import gradio as gr
+
 
 
 codgen_model = os.environ.get('MODEL', 'Salesforce/codegen-350M-mono')
@@ -34,3 +36,6 @@ def generate():
 @app.route('/status', methods=['GET'])
 def status():
     return {'status': 'OK'}
+    
+iface = gr.Interface(fn=generate, inputs="text", outputs="text")
+iface.launch()