Upload app.py

app.py

@@ -1,3 +1,29 @@
+import sys
+import subprocess
+import pkg_resources
+
+def install(package):
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", package])
+
+required_packages = {
+    'gradio': '3.50.2',
+    'torch': '2.0.0',
+    'transformers': '4.35.0',
+    'einops': '0.6.1',
+    'tqdm': '4.66.1',
+    'bitsandbytes': '0.41.1'
+}
+
+for package, version in required_packages.items():
+    try:
+        pkg_resources.require(f"{package}>={version}")
+    except pkg_resources.VersionConflict:
+        print(f"{package} version {version} or higher is required. Upgrading...")
+        install(f"{package}>={version}")
+    except pkg_resources.DistributionNotFound:
+        print(f"{package} is not installed. Installing...")
+        install(f"{package}>={version}")
+
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig