Update server.py

server.py

@@ -322,3 +322,32 @@ def get_grad_attn_matrix(req: GradAttnModelRequest):
         print("SERVER EXCEPTION:", e)
         return {"error": str(e)}
     
+
+
+@app.post("/load_all_files")
+def load_all_files():
+
+    print('load BERTmlm ')
+    BERTVisualizer('mlm')
+    print('load BERTmnli ')
+    BERTVisualizer('mnli')
+    print('load BERTsst ')
+    BERTVisualizer('sst')
+
+    print('load roBERTmlm ')
+    RoBERTaVisualizer('mlm')
+    print('load roBERTmnli')
+
+    RoBERTaVisualizer('mnli')
+    print('load roBERTsst')
+    RoBERTaVisualizer('sst')
+
+    print('load distillBERTmlm ')
+    DistilBERTVisualizer('mlm')
+    print('load distillBERTmmli ')
+    DistilBERTVisualizer('mnli')
+    print('load distillBERTsst ')
+    DistilBERTVisualizer('sst')
+
+
+