MrSimple01 commited on
Commit
4550cfd
·
verified ·
1 Parent(s): b559d3b

Update src/quiz_processing.py

Browse files
Files changed (1) hide show
  1. src/quiz_processing.py +1 -0
src/quiz_processing.py CHANGED
@@ -7,6 +7,7 @@ from typing import Dict, Any, List, Optional
7
  from transformers import AutoTokenizer
8
  from sentence_transformers import SentenceTransformer
9
  from huggingface_hub import login
 
10
 
11
  GEMINI_MODEL = "gemini-2.0-flash"
12
  DEFAULT_TEMPERATURE = 0.7
 
7
  from transformers import AutoTokenizer
8
  from sentence_transformers import SentenceTransformer
9
  from huggingface_hub import login
10
+ from src.prompts import SUMMARY_PROMPT_TEMPLATE, SYSTEM_PROMPT, QUIZ_PROMPT_TEMPLATE
11
 
12
  GEMINI_MODEL = "gemini-2.0-flash"
13
  DEFAULT_TEMPERATURE = 0.7