Update app.py

app.py

@@ -10,6 +10,12 @@ from langchain.chains import RetrievalQA
 from langchain.llms import HuggingFacePipeline
 from transformers import AutoTokenizer, TextStreamer, pipeline, BitsAndBytesConfig, AutoModelForCausalLM
 
+TORCH_VERSION = torch.__version__
+SUPPORTED_TORCH_VERSIONS = ['2.0.1', '2.1.2', '2.2.2', '2.4.0']
+if TORCH_VERSION.rsplit('+')[0] not in SUPPORTED_TORCH_VERSIONS:
+    print(f"Warning: Current PyTorch version {TORCH_VERSION} may not be compatible with ZeroGPU. "
+          f"Supported versions are: {', '.join(SUPPORTED_TORCH_VERSIONS)}")
+
 # Model initialization
 model_id = "meta-llama/Llama-3.2-3B-Instruct"
 token = os.environ.get("HF_TOKEN")