stivenDR14 commited on
Commit
1c7d5c8
·
1 Parent(s): 07c0a81

update model and prompt

Browse files
Files changed (2) hide show
  1. pdf_processor.py +1 -1
  2. utils.py +2 -2
pdf_processor.py CHANGED
@@ -313,7 +313,7 @@ class PDFProcessor:
313
  Ensure clarity, logical flow, and coherence between the combined points.
314
  The summary must be in {language}.
315
  The output must be in markdown format.
316
- Output:
317
  """
318
  )
319
 
 
313
  Ensure clarity, logical flow, and coherence between the combined points.
314
  The summary must be in {language}.
315
  The output must be in markdown format.
316
+ Summary:
317
  """
318
  )
319
 
utils.py CHANGED
@@ -10,7 +10,7 @@ if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
12
  "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
13
- "Huggingface / Google Gemma 3 12B Instruct": "google/gemma-3-12b-it",
14
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
15
  "IBM Granite3.1 dense / Ollama local": "ollama",
16
  "Open AI / GPT-4o-mini": "openai",
@@ -19,7 +19,7 @@ else:
19
  AI_MODELS = {
20
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
21
  "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
22
- "Huggingface / Google Gemma 3 12B Instruct": "google/gemma-3-12b-it",
23
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
24
  "Open AI / GPT-4o-mini": "openai",
25
  }
 
10
  AI_MODELS = {
11
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
12
  "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
13
+ "Huggingface / Google Gemma 3 4B Instruct": "google/gemma-3-4b-it",
14
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
15
  "IBM Granite3.1 dense / Ollama local": "ollama",
16
  "Open AI / GPT-4o-mini": "openai",
 
19
  AI_MODELS = {
20
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
21
  "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
22
+ "Huggingface / Google Gemma 3 4B Instruct": "google/gemma-3-4b-it",
23
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
24
  "Open AI / GPT-4o-mini": "openai",
25
  }