Update app.py

app.py

@@ -6,11 +6,15 @@ import os
 from threading import RLock
 from datetime import datetime
 
+# ---------------------------------
+
 import pkg_resources
 
 for dist in pkg_resources.working_set:
     print(f"{dist.project_name}=={dist.version}")
 
+# ---------------------------------
+
 preSetPrompt = thePrompt
 negPreSetPrompt = ""
 
@@ -34,6 +38,17 @@ HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None #
 api_key=HF_TOKEN
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
+# ---------------------------------
+
+from huggingface_hub import InferenceClient
+
+client = InferenceClient(
+    provider="fal-ai",
+    api_key=HF_TOKEN,
+)
+
+# ---------------------------------
+
 def get_current_time():
     now = datetime.now()
     current_time = now.strftime("%y-%m-%d %H:%M:%S")