oobabooga commited on
Commit
71d0610
·
verified ·
1 Parent(s): 8c5d2a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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: {model_name}"
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()