Anurag Bhardwaj commited on
Commit
3d646b7
·
verified ·
1 Parent(s): de1e61c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -10,7 +10,8 @@ required_packages = {
10
  "torch": "torch",
11
  "diffusers": "diffusers",
12
  "PIL": "pillow",
13
- "spaces": "spaces" # If this is a custom package in your environment.
 
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