TKM03 commited on
Commit
49d71b6
·
verified ·
1 Parent(s): 81798ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -13
app.py CHANGED
@@ -18,12 +18,9 @@ logging.basicConfig(
18
  )
19
  logger = logging.getLogger("CompanyChatbot")
20
 
21
- #Another model
22
- #mistralai/Mixtral-8x7B-Instruct-v0.1
23
-
24
- # Environment variables (Set these in Hugging Face Spaces settings)
25
- HF_MODEL = os.environ.get("HF_MODEL", "MaziyarPanahi/calme-3.2-instruct-78b") # Powerful free model
26
- HF_API_TOKEN = os.environ.get("HF_API_TOKEN") # Must be set in Spaces settings
27
  COMPANY_NAME = os.environ.get("COMPANY_NAME", "AI")
28
  DEFAULT_SYSTEM_PROMPT = os.environ.get("DEFAULT_SYSTEM_PROMPT",
29
  f"You are {COMPANY_NAME}'s professional AI assistant. Be helpful, accurate, and concise.")
@@ -396,10 +393,11 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} AI Assistant") as demo:
396
  outputs=[login_group, chat_group, user_id, user_role]
397
  )
398
 
399
- # Launch the app for Hugging Face Spaces
400
- demo.launch(
401
- server_name="0.0.0.0",
402
- server_port=7860,
403
- share=True,
404
- show_error=True
405
- )
 
 
18
  )
19
  logger = logging.getLogger("CompanyChatbot")
20
 
21
+ # Environment variables
22
+ HF_MODEL = os.environ.get("HF_MODEL", "MaziyarPanahi/calme-3.2-instruct-78b")
23
+ HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
 
 
 
24
  COMPANY_NAME = os.environ.get("COMPANY_NAME", "AI")
25
  DEFAULT_SYSTEM_PROMPT = os.environ.get("DEFAULT_SYSTEM_PROMPT",
26
  f"You are {COMPANY_NAME}'s professional AI assistant. Be helpful, accurate, and concise.")
 
393
  outputs=[login_group, chat_group, user_id, user_role]
394
  )
395
 
396
+ if __name__ == "__main__":
397
+ # Launch the app for Hugging Face Spaces
398
+ demo.launch(
399
+ server_name="0.0.0.0",
400
+ server_port=7860,
401
+ share=True,
402
+ show_error=True
403
+ )