kevinkal commited on
Commit
59699e2
·
verified ·
1 Parent(s): 769f5be

Add gemma models and token statistics

Browse files
Files changed (1) hide show
  1. routers/powerpoint.py +5 -1
routers/powerpoint.py CHANGED
@@ -42,6 +42,8 @@ class MultimodalModel(str, Enum):
42
  gemini_20_flash = "gemini-2.0-flash"
43
  gemini_20_flash_lite = "gemini-2.0-flash-lite"
44
  gemini_25_pro = "gemini-2.5-pro"
 
 
45
 
46
  class TemplateType(str, Enum):
47
  single_graph = "single_graph"
@@ -506,7 +508,7 @@ async def image_to_pptx(
506
  """
507
  try:
508
  # Get available model
509
- model = get_available_model()
510
 
511
  # Get template generator
512
  template_generator = TEMPLATE_GENERATORS.get(template)
@@ -531,6 +533,8 @@ async def image_to_pptx(
531
  'response_schema': template_generator.get_response_schema(),
532
  }
533
  )
 
 
534
 
535
  # Parse the response
536
  presentation_data = template_generator.get_response_schema().model_validate(json.loads(response.text))
 
42
  gemini_20_flash = "gemini-2.0-flash"
43
  gemini_20_flash_lite = "gemini-2.0-flash-lite"
44
  gemini_25_pro = "gemini-2.5-pro"
45
+ gemma_3_4b = "gemma-3-4b-it"
46
+ gemma_3_12b = "gemma-3-12b-it"
47
 
48
  class TemplateType(str, Enum):
49
  single_graph = "single_graph"
 
508
  """
509
  try:
510
  # Get available model
511
+ #model = get_available_model()
512
 
513
  # Get template generator
514
  template_generator = TEMPLATE_GENERATORS.get(template)
 
533
  'response_schema': template_generator.get_response_schema(),
534
  }
535
  )
536
+ # Display input/output tokens
537
+ print(response.usage_metadata)
538
 
539
  # Parse the response
540
  presentation_data = template_generator.get_response_schema().model_validate(json.loads(response.text))