Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ h1 {
|
|
33 |
}
|
34 |
"""
|
35 |
|
36 |
-
|
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(
|