Spaces:
Paused
Paused
Commit
·
2ebe745
1
Parent(s):
4c046f5
another attempt to fix bool error
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ model.eval()
|
|
42 |
|
43 |
# === Role Agent with instruction/input/output format ===
|
44 |
class RoleAgent:
|
45 |
-
def __init__(self, role_instruction, tokenizer
|
46 |
self.tokenizer = tokenizer
|
47 |
self.model = model
|
48 |
self.role_instruction = role_instruction
|
|
|
42 |
|
43 |
# === Role Agent with instruction/input/output format ===
|
44 |
class RoleAgent:
|
45 |
+
def __init__(self, role_instruction, tokenizer, model):
|
46 |
self.tokenizer = tokenizer
|
47 |
self.model = model
|
48 |
self.role_instruction = role_instruction
|