1MR commited on
Commit
60cde18
·
verified ·
1 Parent(s): f0c5978

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -62,11 +62,10 @@ class_labels = {0: 'Baked Potato',1: 'Burger',2: 'Cake',3: 'Chips',4: 'Crispy Ch
62
  api_key='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12'
63
  llm = HuggingFaceEndpoint(
64
  repo_id="Qwen/Qwen2.5-72B-Instruct",
65
- max_new_tokens=512,
66
- client=api_key[:-2]
67
  )
68
  chat = ChatHuggingFace(llm=llm, verbose=True)
69
-
70
  str_output_parser = StrOutputParser()
71
 
72
  chain_label = predict_label
 
62
  api_key='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12'
63
  llm = HuggingFaceEndpoint(
64
  repo_id="Qwen/Qwen2.5-72B-Instruct",
65
+ max_new_tokens=512
 
66
  )
67
  chat = ChatHuggingFace(llm=llm, verbose=True)
68
+ #client=api_key[:-2]
69
  str_output_parser = StrOutputParser()
70
 
71
  chain_label = predict_label