Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -21,7 +21,7 @@ class EndpointHandler():
|
|
21 |
# text = self.prompt_ar.format_map({'Question':inputs})
|
22 |
# the prompt will be preapred on the backend fully
|
23 |
text = inputs
|
24 |
-
print(f"///prompt: text")
|
25 |
|
26 |
input_ids = self.tokenizer(text, return_tensors="pt").input_ids
|
27 |
inputs = input_ids.to(self.device)
|
|
|
21 |
# text = self.prompt_ar.format_map({'Question':inputs})
|
22 |
# the prompt will be preapred on the backend fully
|
23 |
text = inputs
|
24 |
+
print(f"///prompt: {text}")
|
25 |
|
26 |
input_ids = self.tokenizer(text, return_tensors="pt").input_ids
|
27 |
inputs = input_ids.to(self.device)
|