Update server.py

server.py

@@ -20,7 +20,6 @@ import posthog
 from chromadb.config import Settings
 from sentence_transformers import SentenceTransformer
 from werkzeug.middleware.proxy_fix import ProxyFix
-import argparse
 from transformers import AutoTokenizer, AutoProcessor, pipeline
 from transformers import AutoModelForCausalLM, AutoModelForSeq2SeqLM
 from transformers import BlipForConditionalGeneration, GPT2Tokenizer
@@ -31,22 +30,22 @@ from colorama import Fore, Style, init as colorama_init
 
 colorama_init()
 
+
 parser.add_argument(
     "--classification-model", help="Load a custom text classification model"
 )
 
+
+
+
 port = 7860
 host = "0.0.0.0"
 
 summarization_model = (
-    args.summarization_model
-    if args.summarization_model
-    else DEFAULT_SUMMARIZATION_MODEL
+    Qiliang/bart-large-cnn-samsum-ChatGPT_v3
 )
 classification_model = (
-    args.classification_model
-    if args.classification_model
-    else DEFAULT_CLASSIFICATION_MODEL
+    joeddav/distilbert-base-uncased-go-emotions-student
 )