Update app.py

app.py

@@ -27,16 +27,16 @@ def get_model(type='Molecule Function'):
     elif type == 'Biological Process':
         model.load_checkpoint("model/checkpoint_bp1.pth")
         model.to('cuda')
-    elif type == 'Cellar Component':
-        model.load_checkpoint("model/checkpoint_cc2.pth")
-        model.to('cuda')
+    # elif type == 'Cellar Component':
+    #     model.load_checkpoint("model/checkpoint_cc2.pth")
+    #     model.to('cuda')
     return model
 
 
 models = {
     'Molecule Function': get_model('Molecule Function'),
     'Biological Process': get_model('Biological Process'),
-    'Cellar Component': get_model('Cellar Component'),
+    # 'Cellar Component': get_model('Cellar Component'),
     }