Isaac Zepeda commited on
Commit ·
919faf4
1
Parent(s): c2ee745
Testing the handler
Browse files- .gitignore +1 -0
- requirements.txt +1 -0
- test.py +9 -0
.gitignore
CHANGED
|
@@ -1 +1,2 @@
|
|
| 1 |
/**/.DS_Store
|
|
|
|
|
|
| 1 |
/**/.DS_Store
|
| 2 |
+
__pycache__/
|
requirements.txt
CHANGED
|
@@ -1 +1,2 @@
|
|
|
|
|
| 1 |
diffusers==0.7.2
|
|
|
|
| 1 |
+
torch
|
| 2 |
diffusers==0.7.2
|
test.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from handler import EndpointHandler
|
| 2 |
+
|
| 3 |
+
my_handler = EndpointHandler(path=".")
|
| 4 |
+
|
| 5 |
+
prompt = "An astronaut riding horse"
|
| 6 |
+
|
| 7 |
+
input = { "inputs": prompt }
|
| 8 |
+
|
| 9 |
+
print("result: ", my_handler(input))
|