DawnC commited on
Commit
48d17f9
·
verified ·
1 Parent(s): 089125f

Update breed_recommendation_enhanced.py

Browse files
Files changed (1) hide show
  1. breed_recommendation_enhanced.py +3 -1
breed_recommendation_enhanced.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  from typing import Dict, List, Any, Optional
3
  import traceback
 
4
  from semantic_breed_recommender import get_breed_recommendations_by_description, get_enhanced_recommendations_with_unified_scoring
5
  from natural_language_processor import get_nlp_processor
6
  from recommendation_html_format import format_unified_recommendation_html
@@ -477,6 +478,7 @@ def create_recommendation_tab(
477
  print(f"[DESC][VALIDATE][WARN] {str(e)}")
478
  return gr.update(visible=False), True
479
 
 
480
  def find_breeds_by_description(description_text):
481
  """Find breeds based on description and persist history"""
482
  try:
@@ -637,4 +639,4 @@ def create_recommendation_tab(
637
  'criteria_results': locals().get('criteria_results'),
638
  'description_results': locals().get('description_results'),
639
  'description_input': locals().get('description_input')
640
- }
 
1
  import gradio as gr
2
  from typing import Dict, List, Any, Optional
3
  import traceback
4
+ import spaces
5
  from semantic_breed_recommender import get_breed_recommendations_by_description, get_enhanced_recommendations_with_unified_scoring
6
  from natural_language_processor import get_nlp_processor
7
  from recommendation_html_format import format_unified_recommendation_html
 
478
  print(f"[DESC][VALIDATE][WARN] {str(e)}")
479
  return gr.update(visible=False), True
480
 
481
+ @spaces.GPU
482
  def find_breeds_by_description(description_text):
483
  """Find breeds based on description and persist history"""
484
  try:
 
639
  'criteria_results': locals().get('criteria_results'),
640
  'description_results': locals().get('description_results'),
641
  'description_input': locals().get('description_input')
642
+ }