infinitymatter commited on
Commit
8116448
·
verified ·
1 Parent(s): f7da82d

Update src/models.py

Browse files
Files changed (1) hide show
  1. src/models.py +0 -1
src/models.py CHANGED
@@ -27,7 +27,6 @@ claude = InferenceClient(token=anthropic_api_key)
27
  OPENAI_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
28
  CLAUDE_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
29
 
30
- # Call OpenAI's GPT model with prompt and system message
31
  def get_gpt_completion(prompt, system_message):
32
  try:
33
  response = openai.text_generation(prompt=f"{system_message}\n{prompt}", max_new_tokens=200)
 
27
  OPENAI_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
28
  CLAUDE_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
29
 
 
30
  def get_gpt_completion(prompt, system_message):
31
  try:
32
  response = openai.text_generation(prompt=f"{system_message}\n{prompt}", max_new_tokens=200)