Update app.py

app.py

@@ -18,10 +18,10 @@ from transformers import pipeline
 sys.path.append('.')
 sys.path.append('./Time_TravelRephotography')
 from argparse import Namespace
-from projector import (
-    ProjectorArguments,
-    main,
-)
+#from projector import (
+#    ProjectorArguments,
+#    main,
+#)
 sys.path.insert(0, 'StyleGAN-Human')
 
 TITLE = 'Time-TravelRephotography'
@@ -52,6 +52,7 @@ def predict(text):
   return pipe(text)[0]["translation_text"]
  
 def main():
+    tools.setup_helpers.cuda.CUDA_HOME
     load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
     iface = gr.Interface(
       fn=predict,