Clean up

app.py

@@ -21,13 +21,11 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 MODEL_ID_OPT_6_7B = "Salesforce/blip2-opt-6.7b"
 MODEL_ID_FLAN_T5_XXL = "Salesforce/blip2-flan-t5-xxl"
 MODEL_ID = os.getenv("MODEL_ID", MODEL_ID_FLAN_T5_XXL)
+assert MODEL_ID in [MODEL_ID_OPT_6_7B, MODEL_ID_FLAN_T5_XXL]
 
 if torch.cuda.is_available():
     processor = AutoProcessor.from_pretrained(MODEL_ID)
     model = Blip2ForConditionalGeneration.from_pretrained(MODEL_ID, device_map="auto", load_in_8bit=True)
-else:
-    processor = None
-    model = None
 
 
 @spaces.GPU