Remove unused Top-p slider from app.py and adjust top_k parameter in EndToEndRAG class from 5 to 1 for improved performance and simplicity.

app.py

@@ -53,13 +53,6 @@ chatbot = gr.ChatInterface(
         gr.Textbox(value="تو یک دستیار پاسخ‌گوی دقیق به زبان فارسی هستی.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=256, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.0, maximum=2.0, value=0.2, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (unused)",
-        ),
         gr.Textbox(value="", label="Image URL (optional)"),
     ],
 )

end_to_end_class.py

@@ -34,7 +34,7 @@ class EndToEndRAG:
         device: Optional[str] = None,
         text_weight: float = 0.7,
         image_weight: float = 0.3,
-        top_k: int = 5,
+        top_k: int = 1,
         max_new_tokens: int = 10,
         temperature: float = 0.1,
     ) -> None:
@@ -80,7 +80,7 @@ class EndToEndRAG:
         hf_token: Optional[str] = None,
         text_weight: float = 0.7,
         image_weight: float = 0.3,
-        top_k: int = 5,
+        top_k: int = 1,
         max_new_tokens: int = 10,
         temperature: float = 0.1,
         device: Optional[str] = None,