Spaces:
Running
Running
fix transformers js redesign
Browse files- 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",
|