Testing the handler

.gitignore

@@ -1 +1,2 @@
 /**/.DS_Store
+__pycache__/

requirements.txt

@@ -1 +1,2 @@
+torch
 diffusers==0.7.2

test.py

@@ -0,0 +1,9 @@
+from handler import EndpointHandler
+
+my_handler = EndpointHandler(path=".")
+
+prompt = "An astronaut riding horse"
+
+input = { "inputs": prompt }
+
+print("result: ", my_handler(input))