1MR commited on
Commit
3e143b9
·
verified ·
1 Parent(s): bd35f20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -61,11 +61,14 @@ class_labels = {0: 'Baked Potato',1: 'Burger',2: 'Cake',3: 'Chips',4: 'Crispy Ch
61
 
62
  api_key='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00'
63
  llm = HuggingFaceEndpoint(
64
- repo_id="microsoft/Phi-3-mini-4k-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
 
 
61
 
62
  api_key='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00'
63
  llm = HuggingFaceEndpoint(
64
+ repo_id="Qwen/Qwen2.5-72B-Instruct",
65
+ task="text-generation",
66
  max_new_tokens=512,
67
+ do_sample=False,
68
+ repetition_penalty=1.03,
69
  )
70
+
71
+ chat_model = ChatHuggingFace(llm=llm)
72
 
73
  str_output_parser = StrOutputParser()
74