Update app.py

app.py

@@ -29,11 +29,11 @@ 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}',
+    path = hf_hub_download(f'{path}',
+                           f'{file_name}',
                            use_auth_token=TOKEN)
     with open(path, 'rb') as f:
-        model = pickle.load(f)['G_ema']
+        model = torch.load(f)
     model.eval()
     model.to(device)
     with torch.inference_mode():
@@ -44,12 +44,16 @@ def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
    
 def predict(text):
   return pipe(text)[0]["translation_text"]
-  
-iface = gr.Interface(
-  fn=predict, 
-  inputs='text',
-  outputs='text',
-  examples=[["Time-TravelRephotography"]]
-)
-
-iface.launch()
+ 
+def main():
+    iface = gr.Interface(
+      fn=predict, 
+      inputs='text',
+      outputs='text',
+      examples=[["Time-TravelRephotography"]]
+    )
+    
+    iface.launch()
+    
+if __name__ == '__main__':
+    main()