prateekbh commited on
Commit
63db7c6
1 Parent(s): 95b0407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -43,17 +43,17 @@ def getProductDetails(history, image):
43
  clients = InferenceClient("google/gemma-7b")
44
  rand_val = random.randint(1, 1111111111111111)
45
  if not history:
46
- history = [[{"role": "system", "content": "You are a helpful assistant."}]]
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"
57
  # formatted_prompt = self.format_prompt(f"{system_prompt}, {prompt}", history)
58
  # stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=stream_output, details=True, return_full_text=False)
59
  # output = ""
 
43
  clients = InferenceClient("google/gemma-7b")
44
  rand_val = random.randint(1, 1111111111111111)
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"
57
  # formatted_prompt = self.format_prompt(f"{system_prompt}, {prompt}", history)
58
  # stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=stream_output, details=True, return_full_text=False)
59
  # output = ""