call banana

app.py

@@ -1,7 +1,19 @@
 import gradio as gr
+import banana_dev as banana
+import os
+api_key = os.getenv("API_KEY")
+model_key = os.getenv("MODEL_KEY")
+def call_model(prompt):
+    model_parameters = {
+        "prompt": prompt,
+        "max_length":1000
 
-def greet(name):
-    return "Hello " + name + "!!"
+    }
+    out = banana.run(api_key, model_key, model_parameters)
+    if out.message!="success":
+            return "error"
+    
+    return out["modelOutputs"][0]["output"]
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(fn=call_model, inputs="text", outputs="text")
 iface.launch()

requirements.txt

@@ -0,0 +1 @@
+banana-dev