nikravan commited on
Commit
6f11ed6
1 Parent(s): f275cee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ h1 {
33
  }
34
  """
35
 
36
- flag=False
37
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
38
 
39
 
@@ -97,7 +97,7 @@ def mode_load(path):
97
 
98
 
99
  @spaces.GPU()
100
-
101
  def stream_chat(message, history: list, temperature: float, max_length: int, top_p: float, top_k: int, penalty: float):
102
  if flag==False:
103
  model = AutoModelForCausalLM.from_pretrained(
 
33
  }
34
  """
35
 
36
+
37
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
38
 
39
 
 
97
 
98
 
99
  @spaces.GPU()
100
+ flag=False
101
  def stream_chat(message, history: list, temperature: float, max_length: int, top_p: float, top_k: int, penalty: float):
102
  if flag==False:
103
  model = AutoModelForCausalLM.from_pretrained(