monsimas commited on
Commit
9d0a6de
1 Parent(s): b5d1ea0

change temp, top p and max tokens

Browse files
Files changed (1) hide show
  1. app.py +5 -9
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.7):
57
- """
58
- Make a single request to the Mistral model, including both system and user messages,
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: