Update app.py

app.py

@@ -86,7 +86,7 @@ def respond(
     **kwargs,
 ) -> str:
     # Initialize the model
-    model = pipeline("text-generation", model=selected_model)
+    model = pipeline("text-generation", model="enricoros/big-agi")
 
     # Generate a response
     response = model(
@@ -101,43 +101,6 @@ def respond(
     assistant_response = response[0]['generated_text'].strip()
 
     return assistant_response
-class MyChatbot(gr.ChatInterface):
-    def __init__(self, fn, *args, **kwargs):
-        super().__init__(fn, *args, **kwargs)
-
-    def update_chat_history(self, message: str, is_user: bool) -> None:
-        if is_user:
-            self.history.append((message, None))
-        else:
-            self.history.append((None, message))
-
-    def compute(self, *args, **kwargs):
-        command = args[0]
-        history = self.history
-        system_message = self.additional_inputs["system_message"]
-        max_tokens = self.additional_inputs["max_new_tokens"]
-        temperature = self.additional_inputs["temperature"]
-        top_p = self.additional_inputs["top_p"]
-        github_api_token = self.additional_inputs["github_api_token"]
-        github_username = self.additional_inputs["github_username"]
-        github_repository = self.additional_inputs["github_repository"]
-        selected_model = self.additional_inputs["model_dropdown"]
-        severity = self.additional_inputs["severity_dropdown"]
-        programming_language = self.additional_inputs["programming_language_textbox"]
-
-        return self.respond(
-            command,
-            history,
-            system_message,
-            max_tokens,
-            temperature,
-            top_p,
-            github_api_token,
-            github_username,
-            github_repository,
-            selected_model,
-            severity,
-            programming_language,
         )
 
 with gr.Blocks() as demo: