atz21 commited on
Commit
733048e
·
verified ·
1 Parent(s): 28dd385

trying flash

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -99,7 +99,7 @@ def save_as_pdf(text, filename="output.pdf"):
99
  def transcribe_student(ans_file):
100
  try:
101
  ans_uploaded = genai.upload_file(path=ans_file, display_name="Answer Sheet")
102
- model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
103
 
104
  resp = model.generate_content([PROMPTS["TRANSCRIPTION_PROMPT"]["content"], ans_uploaded])
105
  transcription = getattr(resp, "text", None)
@@ -115,7 +115,7 @@ def transcribe_student(ans_file):
115
  def transcribe_ms(ms_file):
116
  try:
117
  ms_uploaded = genai.upload_file(path=ms_file, display_name="Markscheme")
118
- model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
119
 
120
  resp = model.generate_content([PROMPTS["MARKSCHEME_TRANSCRIPTION_PROMPT"]["content"], ms_uploaded])
121
  ms_transcription = getattr(resp, "text", None)
@@ -131,7 +131,7 @@ def transcribe_ms(ms_file):
131
  def grade(qp_file, ms_transcription, student_transcription):
132
  try:
133
  qp_uploaded = genai.upload_file(path=qp_file, display_name="Question Paper")
134
- model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
135
 
136
  response = model.generate_content([
137
  PROMPTS["GRADING_PROMPT"]["content"],
 
99
  def transcribe_student(ans_file):
100
  try:
101
  ans_uploaded = genai.upload_file(path=ans_file, display_name="Answer Sheet")
102
+ model = genai.GenerativeModel("gemini-2.5-flash", generation_config={"temperature": 0})
103
 
104
  resp = model.generate_content([PROMPTS["TRANSCRIPTION_PROMPT"]["content"], ans_uploaded])
105
  transcription = getattr(resp, "text", None)
 
115
  def transcribe_ms(ms_file):
116
  try:
117
  ms_uploaded = genai.upload_file(path=ms_file, display_name="Markscheme")
118
+ model = genai.GenerativeModel("gemini-2.5-flash", generation_config={"temperature": 0})
119
 
120
  resp = model.generate_content([PROMPTS["MARKSCHEME_TRANSCRIPTION_PROMPT"]["content"], ms_uploaded])
121
  ms_transcription = getattr(resp, "text", None)
 
131
  def grade(qp_file, ms_transcription, student_transcription):
132
  try:
133
  qp_uploaded = genai.upload_file(path=qp_file, display_name="Question Paper")
134
+ model = genai.GenerativeModel("gemini-2.5-flash", generation_config={"temperature": 0})
135
 
136
  response = model.generate_content([
137
  PROMPTS["GRADING_PROMPT"]["content"],