Update app.py

app.py

@@ -7,7 +7,7 @@ import spaces
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
 
-subprocess.run('pip install --upgrade "transformers>=4.45" tokenizers', shell=True)
+subprocess.run('pip install transformers==4.45.2 torch==2.4.1', shell=True)
 subprocess.run('pip install --upgrade flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 MODEL_ID = "AtAndDev/marco-qwq-7B"