ysharma HF staff commited on
Commit
dd071e4
1 Parent(s): b791fd8
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,13 +49,13 @@ def text_generate(problem, template, prompt):
49
  {
50
  "top_p": 0.9,
51
  "temperature": 1.1,
52
- #"max_new_tokens": 250,
53
  "return_full_text": False
54
  }, "options":
55
  {
56
  "use_cache": True,
57
  "wait_for_model":True
58
- }}
59
  response = requests.post(API_URL, headers=headers, json=json_)
60
  print(f"Response is : {response}")
61
  output = response.json()
 
49
  {
50
  "top_p": 0.9,
51
  "temperature": 1.1,
52
+ "max_new_tokens": 250,
53
  "return_full_text": False
54
  }, "options":
55
  {
56
  "use_cache": True,
57
  "wait_for_model":True
58
+ },}
59
  response = requests.post(API_URL, headers=headers, json=json_)
60
  print(f"Response is : {response}")
61
  output = response.json()