Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -152,6 +152,9 @@ def load_metadata(model_url, current_metadata):
|
|
| 152 |
# Parse the metadata
|
| 153 |
metadata = load_metadata_from_file(file_obj)
|
| 154 |
|
|
|
|
|
|
|
|
|
|
| 155 |
# Extract model name from URL if it's a Hugging Face URL
|
| 156 |
model_name = model_url
|
| 157 |
if "huggingface.co/" in model_url:
|
|
@@ -169,7 +172,7 @@ def load_metadata(model_url, current_metadata):
|
|
| 169 |
metadata['model_size_mb'] = model_size_mb
|
| 170 |
metadata['loaded'] = True
|
| 171 |
|
| 172 |
-
return metadata, gr.update(value=metadata["n_layers"], maximum=metadata["n_layers"]), f"Metadata loaded successfully for: {
|
| 173 |
|
| 174 |
except Exception as e:
|
| 175 |
error_msg = f"Error loading metadata: {str(e)}"
|
|
@@ -429,4 +432,4 @@ def create_ui():
|
|
| 429 |
if __name__ == "__main__":
|
| 430 |
# Create and launch the app
|
| 431 |
demo = create_ui()
|
| 432 |
-
demo.launch()
|
|
|
|
| 152 |
# Parse the metadata
|
| 153 |
metadata = load_metadata_from_file(file_obj)
|
| 154 |
|
| 155 |
+
# Extract filename from URL
|
| 156 |
+
gguf_filename = model_url.split('/')[-1].split('?')[0] # Remove query parameters if any
|
| 157 |
+
|
| 158 |
# Extract model name from URL if it's a Hugging Face URL
|
| 159 |
model_name = model_url
|
| 160 |
if "huggingface.co/" in model_url:
|
|
|
|
| 172 |
metadata['model_size_mb'] = model_size_mb
|
| 173 |
metadata['loaded'] = True
|
| 174 |
|
| 175 |
+
return metadata, gr.update(value=metadata["n_layers"], maximum=metadata["n_layers"]), f"Metadata loaded successfully for: {gguf_filename}"
|
| 176 |
|
| 177 |
except Exception as e:
|
| 178 |
error_msg = f"Error loading metadata: {str(e)}"
|
|
|
|
| 432 |
if __name__ == "__main__":
|
| 433 |
# Create and launch the app
|
| 434 |
demo = create_ui()
|
| 435 |
+
demo.launch()
|