huytofu92 commited on
Commit
1cb8f4e
·
1 Parent(s): a9b6a0e

Skip bird youtube and also add more allowed imports

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. mini_agents.py +4 -8
app.py CHANGED
@@ -82,8 +82,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None, mock_submission: bool =
82
  if not task_id or question_text is None:
83
  print(f"Skipping item with missing task_id or question: {item}")
84
  continue
85
- # if "youtube" in question_text.lower():
86
- # continue
87
  # if "chess" not in question_text.lower():
88
  # continue
89
  try:
 
82
  if not task_id or question_text is None:
83
  print(f"Skipping item with missing task_id or question: {item}")
84
  continue
85
+ if "youtube" in question_text.lower() and "bird" in question_text.lower():
86
+ continue
87
  # if "chess" not in question_text.lower():
88
  # continue
89
  try:
mini_agents.py CHANGED
@@ -36,19 +36,15 @@ AUTHORIZED_IMPORTS = [
36
  # Audio processing
37
  "wave", "speech_recognition", "pytube", "pytube3", "youtube_dl", "pydub", "pyAudioAnalysis",
38
  # Image/Video processing
39
- "cv2", "pytesseract", "onnxruntime", "PIL", "bs4", "tesseract",
40
  # Data processing
41
- "numpy", "pandas", "sklearn", "scipy", "math",
42
  # File handling
43
  "base64", "io", "json", "os", "pickle",
44
  # Visualization
45
- "pyplot", "matplotlib",
46
- # Machine learning
47
- "hmmlearn",
48
- # Web
49
- "requests",
50
  # Utilities
51
- "logging", "yaml", "datetime", "typing", "markdownify", "chess"
52
  ]
53
 
54
  audio_model = InferenceClientModel(
 
36
  # Audio processing
37
  "wave", "speech_recognition", "pytube", "pytube3", "youtube_dl", "pydub", "pyAudioAnalysis",
38
  # Image/Video processing
39
+ "cv2", "cv2.dnn", "cv2.imread", "pytesseract", "onnxruntime", "PIL", "PIL.Image", "bs4", "tesseract",
40
  # Data processing
41
+ "numpy", "pandas", "sklearn", "scipy", "math", "hmmlearn",
42
  # File handling
43
  "base64", "io", "json", "os", "pickle",
44
  # Visualization
45
+ "pyplot", "matplotlib", "matplotlib.pyplot",
 
 
 
 
46
  # Utilities
47
+ "logging", "yaml", "datetime", "typing", "markdownify", "requests", "chess"
48
  ]
49
 
50
  audio_model = InferenceClientModel(