Update server.py

server.py

@@ -216,20 +216,3 @@ def get_grad_attn_matrix(req: GradAttnModelRequest):
         print("SERVER EXCEPTION:", e)
         return {"error": str(e)}
     
-
-if __name__ == "__main__":
-     
-    print('rim ')
-    BERTVisualizer('mlm')
-    BERTVisualizer('mnli')
-    BERTVisualizer('sst')
-
-
-    RoBERTaVisualizer('mlm')
-    RoBERTaVisualizer('mnli')
-    RoBERTaVisualizer('sst')
-
-
-    DistilBERTVisualizer('mlm')
-    DistilBERTVisualizer('mnli')
-    DistilBERTVisualizer('sst')