Azure99 commited on
Commit
c1f8c68
1 Parent(s): 40d5a5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,8 +67,8 @@ def generate(
67
  max_new_tokens=512, temperature=0.5, top_p=0.85, top_k=50, repetition_penalty=1.05)
68
  llm_result = llm.generate(**generation_kwargs)
69
  llm_result = BOT_PREFIX + tokenizer.decode(llm_result.cpu()[0], skip_special_tokens=True)
70
- expanded_description = json.loads(llm_result)["expanded_description"]
71
  print(llm_result)
 
72
  print(expanded_description)
73
 
74
  seed = random.randint(0, 2147483647)
 
67
  max_new_tokens=512, temperature=0.5, top_p=0.85, top_k=50, repetition_penalty=1.05)
68
  llm_result = llm.generate(**generation_kwargs)
69
  llm_result = BOT_PREFIX + tokenizer.decode(llm_result.cpu()[0], skip_special_tokens=True)
 
70
  print(llm_result)
71
+ expanded_description = json.loads(llm_result)["expanded_description"]
72
  print(expanded_description)
73
 
74
  seed = random.randint(0, 2147483647)