Update app.py

app.py

@@ -27,9 +27,6 @@ IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 CHAT_HISTORY = List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]
 
-def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
-    return [sequence.strip() for sequence in stop_sequences.split(",")] if stop_sequences else None
-
 def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
     if image:
         image_height = int(image.height * IMAGE_WIDTH / image.width)
@@ -47,7 +44,6 @@ def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
         image = Image.open(file).convert('RGB')
         image_preview = preprocess_image(image)
         if image_preview:
-            # Display a preview of the uploaded image
             gr.Image(image_preview).render()
         image_path = cache_pil_image(image)
         chatbot.append(((image_path,), None))
@@ -62,7 +58,6 @@ def bot(
     files: Optional[List[str]],
     temperature: float,
     max_output_tokens: int,
-    stop_sequences: str,
     top_k: int,
     top_p: float,
     chatbot: CHAT_HISTORY
@@ -70,12 +65,10 @@ def bot(
     if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
 
-    # Configurar la API con la clave
     genai.configure(api_key=GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
-        stop_sequences=preprocess_stop_sequences(stop_sequences=stop_sequences),
         top_k=top_k,
         top_p=top_p
     )
@@ -121,12 +114,6 @@ max_output_tokens_component = gr.Slider(
     step=1,
     label="Token limit",
 )
-stop_sequences_component = gr.Textbox(
-    label="Add stop sequence",
-    value="",
-    type="text",
-    placeholder="STOP, END",
-)
 top_k_component = gr.Slider(
     minimum=1,
     maximum=40,
@@ -151,7 +138,6 @@ bot_inputs = [
     upload_button_component,
     temperature_component,
     max_output_tokens_component,
-    stop_sequences_component,
     top_k_component,
     top_p_component,
     chatbot_component
@@ -169,7 +155,6 @@ with gr.Blocks() as demo:
         with gr.Accordion("Parameters", open=False):
             temperature_component.render()
             max_output_tokens_component.render()
-            stop_sequences_component.render()
             with gr.Accordion("Advanced", open=False):
                 top_k_component.render()
                 top_p_component.render()