Spaces:
Sleeping
Sleeping
change temp, top p and max tokens
Browse files
app.py
CHANGED
@@ -53,22 +53,18 @@ def get_api_url_from_process_url(process_url):
|
|
53 |
return None
|
54 |
|
55 |
|
56 |
-
def make_request(client, model, system_message, user_message, temperature=0.
|
57 |
-
"""
|
58 |
-
|
59 |
-
and return the response content.
|
60 |
-
"""
|
61 |
-
print(f"Making request to model with temperature: {temperature}")
|
62 |
messages = [
|
63 |
ChatMessage(role="system", content=system_message),
|
64 |
ChatMessage(role="user", content=user_message)
|
65 |
]
|
66 |
-
response = client.chat(model=model, messages=messages, temperature=temperature)
|
67 |
print(f"Response received from model: {response.choices[0].message.content if response.choices else 'No response'}")
|
68 |
return response.choices[0].message.content if response.choices else ""
|
69 |
|
70 |
|
71 |
-
|
72 |
def parse_category(response, categories):
|
73 |
"""
|
74 |
First, try to parse the response as JSON and look for the 'category' key.
|
@@ -131,7 +127,7 @@ def categorize(proposal, categories, allow_multiple_categories, language='Englis
|
|
131 |
temperature = initial_temperature + attempt * temperature_increment
|
132 |
print(f"Attempt {attempt + 1}/{retries}, using temperature: {temperature}")
|
133 |
|
134 |
-
response = make_request(client, model, system_message, user_message, temperature)
|
135 |
|
136 |
category = extract_json_category(response, categories)
|
137 |
if category:
|
|
|
53 |
return None
|
54 |
|
55 |
|
56 |
+
def make_request(client, model, system_message, user_message, temperature=0.3, top_p=0.6, max_tokens=50):
|
57 |
+
""" Make a single request to the Mistral model, including both system and user messages, and return the response content. """
|
58 |
+
print(f"Making request to model with temperature: {temperature}, top_p: {top_p}, max_tokens: {max_tokens}")
|
|
|
|
|
|
|
59 |
messages = [
|
60 |
ChatMessage(role="system", content=system_message),
|
61 |
ChatMessage(role="user", content=user_message)
|
62 |
]
|
63 |
+
response = client.chat(model=model, messages=messages, temperature=temperature, top_p=top_p, max_tokens=max_tokens)
|
64 |
print(f"Response received from model: {response.choices[0].message.content if response.choices else 'No response'}")
|
65 |
return response.choices[0].message.content if response.choices else ""
|
66 |
|
67 |
|
|
|
68 |
def parse_category(response, categories):
|
69 |
"""
|
70 |
First, try to parse the response as JSON and look for the 'category' key.
|
|
|
127 |
temperature = initial_temperature + attempt * temperature_increment
|
128 |
print(f"Attempt {attempt + 1}/{retries}, using temperature: {temperature}")
|
129 |
|
130 |
+
response = make_request(client, model, system_message, user_message, temperature=0.3, top_p=0.6, max_tokens=50)
|
131 |
|
132 |
category = extract_json_category(response, categories)
|
133 |
if category:
|