diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py index 7fb6e561406c9d731d30a83167d47ce9cbec086e..2ba12c0736f796fd23b232030b58eab77ce29cf1 100644 --- a/src/controllers/render_controller.py +++ b/src/controllers/render_controller.py @@ -1,5 +1,5 @@ import os -from gradio_client import Client, handle_file +from gradio_client import Client class RenderController: def get_render(self, hf_token='', text="Hello!!", diff --git a/src/controllers/suggest_controller.py b/src/controllers/suggest_controller.py index 202da39a86758ff8627937ac07a142f239d5e8da..204017d7a8215aa2bb7519a2cf6d1fcec8f7e88f 100644 --- a/src/controllers/suggest_controller.py +++ b/src/controllers/suggest_controller.py @@ -1,5 +1,5 @@ import os -from gradio_client import Client, handle_file +from gradio_client import Client class SuggestController: def get_suggest(self, hf_token='', message='', system_prompt='', temperature=.0,