Hugging Face's logo

Spaces:
StevenChen16
/
AI-Lawyer
Runtime error

App Files Community
6
StevenChen16 commited on
Commit
0f1f41b
·
verified ·
1 Parent(s): ca59e02

update model.

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,8 +4,6 @@ from threading import Thread
4
  from llamafactory.chat import ChatModel
5
  from llamafactory.extras.misc import torch_gc
6
 
7
- # Set an environment variable
8
- HF_TOKEN = os.environ.get("HF_TOKEN", None)
9
 
10
  DESCRIPTION = '''
11
  <div>
@@ -40,7 +38,7 @@ h1 {
40
  """
41
 
42
  args = dict(
43
- model_name_or_path="StevenChen16/llama3-8b-Lawyer",
44
  template="llama3",
45
  finetuning_type="lora",
46
  quantization_bit=8,
 
4
  from llamafactory.chat import ChatModel
5
  from llamafactory.extras.misc import torch_gc
6
 
 
 
7
 
8
  DESCRIPTION = '''
9
  <div>
 
38
  """
39
 
40
  args = dict(
41
+ model_name_or_path="StevenChen16/llama3-8b-Lawyer-chat-v1.5",
42
  template="llama3",
43
  finetuning_type="lora",
44
  quantization_bit=8,