Andre commited on
Commit
2a42615
·
1 Parent(s): ef908f6

“Update”

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -11,6 +11,11 @@ if src_path not in sys.path:
11
  # Import gradio_interface
12
  from gradio_interface import demo
13
  from config.config import api_token
 
 
 
 
 
14
 
15
 
16
  if __name__ == "__main__":
 
11
  # Import gradio_interface
12
  from gradio_interface import demo
13
  from config.config import api_token
14
+ from huggingface_hub import InferenceClient
15
+ from config.models import models
16
+
17
+ # Initialize the InferenceClient with the default model
18
+ client = InferenceClient(models[0]["name"], token=api_token)
19
 
20
 
21
  if __name__ == "__main__":