akhaliq HF Staff commited on
Commit
8021d0b
·
1 Parent(s): c8820b9

fix transformers js redesign

Browse files
Files changed (1) hide show
  1. project_importer.py +11 -3
project_importer.py CHANGED
@@ -110,7 +110,7 @@ class ProjectImporter:
110
 
111
  if files:
112
  code = self._format_multi_file_content(files, username, project_name, space_info.sdk)
113
- language = self._detect_language_from_sdk(space_info.sdk)
114
 
115
  return {
116
  "status": "success",
@@ -513,8 +513,16 @@ Files: {len(files)}
513
 
514
  return patterns.get(sdk, ["app.py", "main.py", "index.html"])
515
 
516
- def _detect_language_from_sdk(self, sdk: str) -> str:
517
- """Detect language/framework from SDK"""
 
 
 
 
 
 
 
 
518
  sdk_map = {
519
  "gradio": "gradio",
520
  "streamlit": "streamlit",
 
110
 
111
  if files:
112
  code = self._format_multi_file_content(files, username, project_name, space_info.sdk)
113
+ language = self._detect_language_from_sdk(space_info.sdk, files) # Pass files for detection
114
 
115
  return {
116
  "status": "success",
 
513
 
514
  return patterns.get(sdk, ["app.py", "main.py", "index.html"])
515
 
516
+ def _detect_language_from_sdk(self, sdk: str, files: Optional[Dict[str, str]] = None) -> str:
517
+ """Detect language/framework from SDK and optionally file contents"""
518
+ # For static SDK, check if it's a transformers.js space by examining files
519
+ if sdk == "static" and files:
520
+ # Check if any JS file contains transformers.js imports
521
+ for filename, content in files.items():
522
+ if filename.endswith(('.js', '.mjs')) or filename == 'index.html':
523
+ if content and ('@xenova/transformers' in content or '@huggingface/transformers' in content):
524
+ return "transformers.js"
525
+
526
  sdk_map = {
527
  "gradio": "gradio",
528
  "streamlit": "streamlit",