Update app.py

app.py

@@ -90,7 +90,7 @@ class SingleDataset(Dataset):
         return features, label
 
 
-def test(model_path, var_path, csv_path, pos):
+def test(use_default, csv_path, pos):
 
     attrib_feature = ['Current', 'Voltage', 'Environment_Temperature', 'Cell_Temperature', 'SOC']
     attrib_label = ['SOH']
@@ -105,16 +105,24 @@ def test(model_path, var_path, csv_path, pos):
     batch_size = 1
     C_rated = 1.1
 
-    model_path = model_path.name
-    var_path = var_path.name
-    csv_path = csv_path.name
+    base = './测试用例/'
+    model_path = base + 'model.pt'
+    var_path = base + 'variables.pkl'
+    if use_default:
+        csv_path = base + 'Test_Data_0.5_3C.csv'
+    else:
+        csv_path =  csv_path.name
+    # model_path = model_path.name
+    # var_path = var_path.name
+    # csv_path = csv_path.name
 
     with open(var_path, 'rb') as f:
         min_val, max_val = pickle.load(f)
 
 
-
+    global max_cycle_index
     test_data = pd.read_csv(csv_path)
+    max_cycle_index = test_data['Cycle_Index'].max()
     test_data['SOH'] = test_data['cycle capacity'] / test_data['cycle capacity'].values[0]
     # C_val = val_data[attrib_label].values[0]
     # scaler_val = train_dataset.scaler
@@ -171,12 +179,12 @@ def test(model_path, var_path, csv_path, pos):
 
         return results
 
+default_checkbox = Checkbox(label="使用默认文件", default=False)
     #
 inputs = [
-    File(label="上传预训练模型"),
-    File(label="上传参数文件"),
+    default_checkbox,
     File(label="上传CSV测试数据"),
-    Number(label="选择循环次数（1~4004）")
+    Number(label='选择循环次数（1~'+str(max_cycle_index)+')')
 ]
 
 outputs = [