Update src/app/response.py

src/app/response.py

@@ -9,11 +9,7 @@ from src.config import (
     model_name,
     sampling,
     stream,
-    top_p,
-    top_k,
-    temperature,
     repetition_penalty,
-    max_new_tokens,
 )
 from src.app.model import load_model_and_tokenizer
 from src.logger import logging
@@ -25,13 +21,24 @@ model, tokenizer, processor = load_model_and_tokenizer(model_name, device)
 
 
 @spaces.GPU(duration=120)
-def describe_image(image: str, question: str) -> str:
+def describe_image(
+    image: str, 
+    question: str,
+    temperature: float,
+    top_p: float,
+    top_k: int,
+    max_new_tokens: int,
+) -> str:
     """
     Generates an answer to a given question based on the provided image and question.
 
     Args:
         - image (str): The path to the image file.
         - question (str): The question text.
+        - temperature (float): The temperature parameter for the model.
+        - top_p (float): The top_p parameter for the model.
+        - top_k (int): The top_k parameter for the model.
+        - max_new_tokens (int): The max tokens to be generated by the model.
 
     Returns:
         str: The generated answer to the question.