johann22 commited on
Commit
615f8ac
·
1 Parent(s): b881441

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -72,7 +72,6 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
72
  print(datetime.datetime.now())
73
  uid=uuid.uuid4()
74
 
75
- seed = random.randint(1,1111111111111111)
76
 
77
  agent=prompts.AI_REPORT_WRITER
78
  ''' if agent_name == "WEB_DEV":
@@ -97,15 +96,18 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
97
  temperature = 1e-2
98
  top_p = float(top_p)
99
 
100
- generate_kwargs = dict(
101
- temperature=temperature,
102
- max_new_tokens=1048,
103
- top_p=top_p,
104
- repetition_penalty=repetition_penalty,
105
- do_sample=True,
106
- seed=seed,
107
- )
108
  while True:
 
 
 
 
 
 
 
 
 
 
109
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
110
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
111
  output = ""
 
72
  print(datetime.datetime.now())
73
  uid=uuid.uuid4()
74
 
 
75
 
76
  agent=prompts.AI_REPORT_WRITER
77
  ''' if agent_name == "WEB_DEV":
 
96
  temperature = 1e-2
97
  top_p = float(top_p)
98
 
99
+
 
 
 
 
 
 
 
100
  while True:
101
+ seed = random.randint(1,1111111111111111)
102
+
103
+ generate_kwargs = dict(
104
+ temperature=temperature,
105
+ max_new_tokens=1048,
106
+ top_p=top_p,
107
+ repetition_penalty=repetition_penalty,
108
+ do_sample=True,
109
+ seed=seed,
110
+ )
111
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
112
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
113
  output = ""