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))
|