fix int issues

app.py

@@ -15,10 +15,10 @@ logging.basicConfig(level=logging.INFO)
 
 # CONSTANTS
 # Get max new tokens from environment variable, if it is not set, default to 2048
-MAX_NEW_TOKENS: int = os.getenv("MAX_NEW_TOKENS", 2048)
+MAX_NEW_TOKENS: int = int(os.getenv("MAX_NEW_TOKENS", 2048))
 
 # Get max number of images to be passed in the prompt
-MAX_NUM_IMAGES: int = os.getenv("MAX_NUM_IMAGES")
+MAX_NUM_IMAGES: int = int(os.getenv("MAX_NUM_IMAGES"))
 if not MAX_NUM_IMAGES:
     raise EnvironmentError("MAX_NUM_IMAGES is not set. Please set it to 1 or more.")
 

gateway.py

@@ -56,7 +56,7 @@ def request_generation(
             json=payload,
             verify=False,
         )
-        print(response.text)
+
         response.raise_for_status()
 
         # Append the conversation ID with the key X-Conversation-ID to the header