fix

app.py

@@ -111,6 +111,13 @@ def save_model(model, model_name, quantization_type, threshold, quant_type_4, do
         )
     return f'<h1> 🤗 DONE</h1><br/>Find your repo here: <a href="https://huggingface.co/{repo_name}" target="_blank" style="text-decoration:underline">{repo_name}</a>'
 
+def is_float(value):
+    try:
+        float(value)
+        return True
+    except ValueError:
+        return False
+
 def quantize_and_save(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None, model_name, quantization_type, threshold, quant_type_4, double_quant_4, quantized_model_name):
     if oauth_token is None : 
         return "Error : Please Sign In to your HuggingFace account to use the quantizer"
@@ -120,10 +127,10 @@ def quantize_and_save(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToke
     if exists_message : 
         return exists_message
 
-    if not threshold.isdigit() : 
-        return "Threshold must be a number"
+    if not is_float(threshold) : 
+        return "Threshold must be a float"
     
-    threshold = int(threshold)
+    threshold = float(threshold)
 
     try:
         quantized_model = quantize_model(model_name, quantization_type, threshold, quant_type_4, double_quant_4, oauth_token, profile.username)