Raiff1982's picture
Update main.py
5958e3a verified
raw
history blame contribute delete
607 Bytes
import logging
import subprocess
import sys
subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade",
"diffusers==0.25.0",
"huggingface_hub==0.20.3",
"transformers==4.37.2",
"torch>=2.1.0"])
from final_codettes_chatbot import HuggingFaceChatbot
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
try:
HuggingFaceChatbot().launch()
except Exception as e:
logging.exception("Startup failed: %s", e)