Update interfaces/illframes.py

interfaces/illframes.py

@@ -64,7 +64,7 @@ def predict(text, model_id, tokenizer_id, label_names):
         for key, value in disk_space.items():
             print(f"{key}: {value}")
 
-        shutil.rmtree("/data/")
+        shutil.rmtree("/data")
         model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", token=HF_TOKEN, force_download=True)
         
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)