Update maker.py

maker.py

@@ -69,33 +69,23 @@ def post_request_beta(payload):
     response.raise_for_status()  # Will raise an HTTPError if the HTTP request returned an unsuccessful status code
     return response.json()
 
-
-def predict_beta(message, chatbot=[], system_prompt=system_prompt):
+def predict_beta(message, chatbot=[], system_prompt=system_prompt, max_new_tokens=1200, temperature=0.4, top_p=0.9, repetition_penalty=0.5, advanced=False):
     client = Client(tulu)
 
     try:
-        # Adjust these parameters as needed
-        max_new_tokens = 1200
-        temperature = 0.4
-        top_p = 0.9
-        repetition_penalty = 0.5
-        advanced = False
-
-        # Making the prediction
         result = client.predict(
-            message,  # Your Message
-            system_prompt,  # Optional Tulu Assistant Message (can adjust if needed)
-            max_new_tokens,
-            temperature,
-            top_p,
-            repetition_penalty,
-            advanced,
+            message,  
+            system_prompt,  
+            max_new_tokens,  
+            temperature, 
+            top_p,  
+            repetition_penalty,  
+            advanced,  
             fn_index=0
         )
 
-        # Extracting the response
         if result is not None and len(result) > 0:
-            bot_message = result[0]  # Assuming the response is in the first element
+            bot_message = result[0]  
             return bot_message
         else:
             raise gr.Error("No response received from the model.")