Update app.py

app.py

@@ -27,9 +27,23 @@ TOKEN = "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"
 
 pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
 
+def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
+    path = hf_hub_download('hysts/StyleGAN-Human',
+                           f'models/{file_name}',
+                           use_auth_token=TOKEN)
+    with open(path, 'rb') as f:
+        model = pickle.load(f)['G_ema']
+    model.eval()
+    model.to(device)
+    with torch.inference_mode():
+        z = torch.zeros((1, model.z_dim)).to(device)
+        label = torch.zeros([1, model.c_dim], device=device)
+        model(z, label, force_fp32=True)
+    return model
+   
 def predict(text):
   return pipe(text)[0]["translation_text"]
-
+  
 iface = gr.Interface(
   fn=predict, 
   inputs='text',