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()
|