Update app.py

app.py

@@ -1,4 +1,14 @@
-from transformers.tools.base import launch_gradio_demo
 from text_to_image import TextToImageTool
 
-launch_gradio_demo(TextToImageTool)
+tool = tool_class()
+
+def fn(*args, **kwargs):
+    return tool(*args, **kwargs)
+
+gr.Interface(
+    fn=fn,
+    inputs=tool_class.inputs,
+    outputs=tool_class.outputs,
+    title=tool_class.__name__,
+    article=tool.description,
+).queue(concurrency_count=5).launch()