Salt commited on
Commit
a7c6cad
·
1 Parent(s): d69c27c

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +1 -17
server.py CHANGED
@@ -54,26 +54,10 @@ class SplitArgs(argparse.Action):
54
  parser = argparse.ArgumentParser(
55
  prog="TavernAI Extras", description="Web API for transformers models"
56
  )
57
- parser.add_argument(
58
- "--port", type=int, help="Specify the port on which the application is hosted"
59
- )
60
- parser.add_argument(
61
- "--listen", action="store_true", help="Host the app on the local network"
62
- )
63
- parser.add_argument(
64
- "--share", action="store_true", help="Share the app on CloudFlare tunnel"
65
- )
66
- parser.add_argument("--cpu", action="store_true", help="Run the models on the CPU")
67
  parser.add_argument("--summarization-model", help="Load a custom summarization model")
68
  parser.add_argument(
69
  "--classification-model", help="Load a custom text classification model"
70
- )
71
- parser.add_argument("--captioning-model", help="Load a custom captioning model")
72
- parser.add_argument(
73
- "--keyphrase-model", help="Load a custom keyphrase extraction model"
74
- )
75
- parser.add_argument("--prompt-model", help="Load a custom prompt generation model")
76
- parser.add_argument("--embedding-model", help="Load a custom text embedding model")
77
 
78
  parser.add_argument(
79
  "--enable-modules",
 
54
  parser = argparse.ArgumentParser(
55
  prog="TavernAI Extras", description="Web API for transformers models"
56
  )
57
+
 
 
 
 
 
 
 
 
 
58
  parser.add_argument("--summarization-model", help="Load a custom summarization model")
59
  parser.add_argument(
60
  "--classification-model", help="Load a custom text classification model"
 
 
 
 
 
 
 
61
 
62
  parser.add_argument(
63
  "--enable-modules",