Commit
·
eb993d1
1
Parent(s):
86cb956
input = data[inputs]
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -8,6 +8,6 @@ class EndpointHandler():
|
|
8 |
self.pipeline.load_lora_weights(".", weight_name="/repository/safetensors.safetensors")
|
9 |
|
10 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
11 |
-
|
12 |
-
result = self.pipeline(
|
13 |
return result
|
|
|
8 |
self.pipeline.load_lora_weights(".", weight_name="/repository/safetensors.safetensors")
|
9 |
|
10 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
11 |
+
input = data['inputs']
|
12 |
+
result = self.pipeline(input)
|
13 |
return result
|