stivenDR14 commited on
Commit
daf9af3
·
1 Parent(s): dd1aad4

updating models

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. utils.py +6 -6
app.py CHANGED
@@ -6,7 +6,7 @@ class PDFProcessorUI:
6
  def __init__(self):
7
  self.processor = PDFProcessor()
8
  self.current_language = "Español"
9
- self.current_ai_model = "Huggingface / Meta Llama 3.2 3B Instruct"
10
  self.current_type_model = "Api Key"
11
 
12
  def change_language(self, language):
 
6
  def __init__(self):
7
  self.processor = PDFProcessor()
8
  self.current_language = "Español"
9
+ self.current_ai_model = "Huggingface / Microsoft Phi 3.5 medium Instruct"
10
  self.current_type_model = "Api Key"
11
 
12
  def change_language(self, language):
utils.py CHANGED
@@ -8,20 +8,20 @@ ENVIRONMENT = os.getenv("ENVIRONMENT")
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
- "Huggingface / Microsoft Phi 3.5 mini Instruct": "microsoft/Phi-3.5-mini-instruct",
12
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
13
- "Huggingface / Google Gemma 2 2B Instruct": "google/gemma-2-2b-it",
14
- "Huggingface / Meta Llama 3.2 3B Instruct": "meta-llama/Llama-3.2-3B-Instruct",
15
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
16
  "IBM Granite3.1 dense / Ollama local": "ollama",
17
  "Open AI / GPT-4o-mini": "openai",
18
  }
19
  else:
20
  AI_MODELS = {
21
- "Huggingface / Microsoft Phi 3.5 mini Instruct": "microsoft/Phi-3.5-mini-instruct",
22
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
23
- "Huggingface / Google Gemma 2 2B Instruct": "google/gemma-2-2b-it",
24
- "Huggingface / Meta Llama 3.2 3B Instruct": "meta-llama/Llama-3.2-3B-Instruct",
25
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
26
  "Open AI / GPT-4o-mini": "openai",
27
  }
 
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
+ "Huggingface / Microsoft Phi 3.5 medium Instruct": "microsoft/Phi-3-medium-128k-instruct",
12
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
13
+ "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
14
+ "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
15
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
16
  "IBM Granite3.1 dense / Ollama local": "ollama",
17
  "Open AI / GPT-4o-mini": "openai",
18
  }
19
  else:
20
  AI_MODELS = {
21
+ "Huggingface / Microsoft Phi 3.5 medium Instruct": "microsoft/Phi-3-medium-128k-instruct",
22
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
23
+ "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
24
+ "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
25
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
26
  "Open AI / GPT-4o-mini": "openai",
27
  }