Update server.py
Browse files
server.py
CHANGED
|
@@ -6,6 +6,9 @@ import io
|
|
| 6 |
import os
|
| 7 |
from PIL import Image
|
| 8 |
from diffusers import StableDiffusionPipeline
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
# Define the MIDM model
|
| 11 |
class MIDM(nn.Module):
|
|
|
|
| 6 |
import os
|
| 7 |
from PIL import Image
|
| 8 |
from diffusers import StableDiffusionPipeline
|
| 9 |
+
import os
|
| 10 |
+
|
| 11 |
+
token = os.getenv("HF_TOKEN")
|
| 12 |
|
| 13 |
# Define the MIDM model
|
| 14 |
class MIDM(nn.Module):
|