Update generate.py

generate.py

@@ -1,6 +1,7 @@
 import argparse
 import os
 import torch
+import subprocess
 from huggingface_hub import snapshot_download
 
 # Arguments