Naman Gupta commited on
Commit
6b72bd2
·
1 Parent(s): d29e78c

Updated api base url

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -6,7 +6,7 @@ from openai import OpenAI
6
  logging.basicConfig(level=logging.INFO)
7
  logger = logging.getLogger(__name__)
8
 
9
- API_BASE_URL = os.environ.get("API_BASE_URL", "http://localhost:7860")
10
  MODEL_NAME = os.environ.get("MODEL_NAME", "mistralai/Mistral-7B-Instruct-v0.3")
11
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
12
 
 
6
  logging.basicConfig(level=logging.INFO)
7
  logger = logging.getLogger(__name__)
8
 
9
+ API_BASE_URL = os.environ.get("API_BASE_URL", "https://rayugacodes-breach-os.hf.space")
10
  MODEL_NAME = os.environ.get("MODEL_NAME", "mistralai/Mistral-7B-Instruct-v0.3")
11
  HF_TOKEN = os.environ.get("HF_TOKEN", "")
12