Update app.py

app.py

@@ -1,4 +1,7 @@
 import gradio as gr
+import torch
+
+pipe = torch.hub.load('nateraw/image-generation:main', 'styleganv3', force_reload=True)
 
 def fn(text):
     return text