Update app.py

app.py

@@ -7,14 +7,13 @@ import os
 import matplotlib.pyplot as plt
 from io import BytesIO
 import base64
-from transformers.models.auto import AutoModel
-from transformers.modeling_utils import PreTrainedModel
-import torch
-from torch.nn.utils import prune
+
+# Choose your backend (PyTorch, TensorFlow, or Flax)
+import torch  # If using PyTorch
 
 # Function to fetch open-weight LLM models
 def fetch_open_weight_models():
-    models = list_models(filter="open-weight", sort="downloads", limit=12)
+    models = list_models(filter="open-weight", sort="downloads", limit=10)
     return [model["id"] for model in models]
 
 # Function to prune a model using the "merge-kit" approach