prateekbh commited on
Commit
f617eac
·
verified ·
1 Parent(s): 63db7c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -45,12 +45,12 @@ def getProductDetails(history, image):
45
  if not history:
46
  history = [[{"role": "system", "content": "You are a helpful assistant."}], [{"role": "user", "content": "hello"}]]
47
  generate_kwargs = dict(
48
- temperature=temp,
49
- max_new_tokens=tokens,
50
- top_p=top_p,
51
- repetition_penalty=rep_p,
52
  do_sample=True,
53
- seed=seed,
54
  )
55
  system_prompt="you're a helpful e-commerce marketting assitant"
56
  prompt="Write me a poem"
 
45
  if not history:
46
  history = [[{"role": "system", "content": "You are a helpful assistant."}], [{"role": "user", "content": "hello"}]]
47
  generate_kwargs = dict(
48
+ temperature=0.67,
49
+ max_new_tokens=1024,
50
+ top_p=0.9,
51
+ repetition_penalty=1,
52
  do_sample=True,
53
+ seed=rand_val,
54
  )
55
  system_prompt="you're a helpful e-commerce marketting assitant"
56
  prompt="Write me a poem"