error handling

app.py

@@ -1,5 +1,11 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+try:
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+except ImportError:
+    import os
+    os.system("pip install transformers torch gradio")
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+    
 import torch
 
 # Load model + tokenizer