prateekbh commited on
Commit
9944c4b
·
verified ·
1 Parent(s): 35213d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,10 +39,10 @@ class StopOnTokens(StoppingCriteria):
39
 
40
  def getProductDetails(history, image):
41
  product_description=getImageDescription(image)
42
- clients = InferenceClient("google/gemma-7b")
43
  # rand_val = random.randint(1, 1111111111111111)
44
  if not history:
45
- history = []
46
  # generate_kwargs = dict(
47
  # temperature=temp,
48
  # max_new_tokens=tokens,
 
39
 
40
  def getProductDetails(history, image):
41
  product_description=getImageDescription(image)
42
+ # clients = InferenceClient("google/gemma-7b")
43
  # rand_val = random.randint(1, 1111111111111111)
44
  if not history:
45
+ history = [{"role": "system", "content": "You are a helpful assistant."}]]
46
  # generate_kwargs = dict(
47
  # temperature=temp,
48
  # max_new_tokens=tokens,