update compressor

.ipynb_checkpoints/app-checkpoint.py

@@ -159,7 +159,7 @@ def main_interface(model_name, sparsity, action):
         compressed_size = get_model_size(compressed_path)
         size_plot = create_size_comparison_plot(original_size, compressed_size)
         
-        return compressed_path, size_plot
+        return size_plot
     finally:
         # Clean up temporary file
         if os.path.exists(temp_path):

app.py

@@ -159,7 +159,7 @@ def main_interface(model_name, sparsity, action):
         compressed_size = get_model_size(compressed_path)
         size_plot = create_size_comparison_plot(original_size, compressed_size)
         
-        return compressed_path, size_plot
+        return size_plot
     finally:
         # Clean up temporary file
         if os.path.exists(temp_path):