Experimenting with library installation

app.py

@@ -1,5 +1,10 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+import spaces
+import subprocess
+# from transformers import AutoTokenizer, AutoModelForCausalLM
+
+result = subprocess.run(["python3", "-m", "pip", "install", "transformers==4.34.0"], shell=True, capture_output=True, text=True)
+print(result.stdout)
 
 model = None
 
@@ -7,9 +12,8 @@ model = None
 def greet(name):
     return "Hello " + name + "!!"
 
-
-def main():
-
+@spaces.GPU
+def load_model():
     print(f"Loading model...")
     model_path = "meta-llama/Meta-Llama-3-8B"
     tokenizer = AutoTokenizer.from_pretrained(model_path)
@@ -19,6 +23,11 @@ def main():
         device_map="auto",
     )
 
+
+def main():
+
+    load_model()
+
     demo = gr.Interface(fn=greet, inputs="text", outputs="text")
     demo.launch()