Rename filehandler.py to handler.py
Browse files
filehandler.py → handler.py
RENAMED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
2 |
|
3 |
-
class
|
4 |
def __init__(self, model_path):
|
5 |
self.model_path = model_path
|
6 |
self.tokenizer = AutoTokenizer.from_pretrained(model_path)
|
@@ -38,4 +38,4 @@ class FileHandler:
|
|
38 |
|
39 |
return response
|
40 |
|
41 |
-
handler =
|
|
|
1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
2 |
|
3 |
+
class Handler:
|
4 |
def __init__(self, model_path):
|
5 |
self.model_path = model_path
|
6 |
self.tokenizer = AutoTokenizer.from_pretrained(model_path)
|
|
|
38 |
|
39 |
return response
|
40 |
|
41 |
+
handler = Handler(".")
|