Spaces:
Sleeping
Sleeping
Anurag Bhardwaj
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,8 @@ required_packages = {
|
|
10 |
"torch": "torch",
|
11 |
"diffusers": "diffusers",
|
12 |
"PIL": "pillow",
|
13 |
-
"spaces": "spaces" #
|
|
|
14 |
}
|
15 |
|
16 |
def install_package(package_name):
|
@@ -22,6 +23,9 @@ for mod, pkg in required_packages.items():
|
|
22 |
print(f"Module {mod} not found, installing {pkg}...")
|
23 |
install_package(pkg)
|
24 |
|
|
|
|
|
|
|
25 |
import random
|
26 |
import gradio as gr
|
27 |
import numpy as np
|
|
|
10 |
"torch": "torch",
|
11 |
"diffusers": "diffusers",
|
12 |
"PIL": "pillow",
|
13 |
+
"spaces": "spaces", # if this is a package available in your environment
|
14 |
+
"transformers": "transformers",
|
15 |
}
|
16 |
|
17 |
def install_package(package_name):
|
|
|
23 |
print(f"Module {mod} not found, installing {pkg}...")
|
24 |
install_package(pkg)
|
25 |
|
26 |
+
# Explicitly import transformers to ensure it's loaded.
|
27 |
+
import transformers
|
28 |
+
|
29 |
import random
|
30 |
import gradio as gr
|
31 |
import numpy as np
|