Update helper_revised.py

helper_revised.py

@@ -50,8 +50,8 @@ class Queries(BaseModel):
     queries: List[str]
     
 class ewriter:
-    def __init__(self, model_names =  ["GPT-3.5","GPT-3.5","GPT-3.5"]):
-    # def __init__(self, model_names =  ["GPT-3.5","Claude 3 Sonnet","GPT-3.5"]):
+    # def __init__(self, model_names =  ["GPT-3.5","GPT-3.5","GPT-3.5"]):
+    def __init__(self, model_names =  ["GPT-3.5","Claude 3 Sonnet","Claude 3.5 Sonnet"]):
         
         models = [self.create_model(model_name) for model_name in model_names]
         
@@ -197,7 +197,8 @@ import time
 
 class writer_gui:
     
-    def __init__(self, model_names=["GPT-3.5","GPT-3.5","GPT-3.5"], share=False):
+    # def __init__(self, model_names=["GPT-3.5","GPT-3.5","GPT-3.5"], share=False):
+    def __init__(self, model_names=["GPT-3.5","Claude 3 Sonnet","Claude 3.5 Sonnet"], share=False):
     # def __init__(self, share=False):
 
         # self.is_new_session = True  # Indicates if it's a new session
@@ -438,8 +439,8 @@ class writer_gui:
                 return (
                 gr.update(value="", interactive=True),  # Reset live output
                 gr.update(value="GPT-3.5", interactive=True),  # Reset model dropdown1
-                gr.update(value="GPT-3.5", interactive=True),  # Reset model dropdown2
-                gr.update(value="GPT-3.5", interactive=True),  # Reset model dropdown3
+                gr.update(value="Claude 3 Sonnet", interactive=True),  # Reset model dropdown2
+                gr.update(value="Claude 3.5 Sonnet", interactive=True),  # Reset model dropdown3
                 gr.update(value="All work and no play makes Jack a Jackass!", interactive=True),  # Reset topic box
                 gr.update(interactive=False),  # Set generate button to inactive
                 gr.update(interactive=False),  # Set continue button to inactive