Update app.py

app.py

@@ -11,8 +11,8 @@ from huggingface_hub import login
 
 login(token=os.environ["HF_TOKEN"])
 
-MODEL_ID = os.environ.get("MODEL_ID", "llava-hf/llava-v1.6-mistral-7b-hf")
-REVISION = os.environ.get("MODEL_REVISION", "main")
+MODEL_ID = os.environ["MODEL_ID"]
+REVISION = os.environ["MODEL_REVISION"]
 processor = LlavaProcessor.from_pretrained(MODEL_ID, revision=REVISION)
 
 model = LlavaForConditionalGeneration.from_pretrained(MODEL_ID, revision=REVISION, torch_dtype=torch.float16, low_cpu_mem_usage=True)