Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoConfig, AutoModelForCausalLM
 from janus.models import MultiModalityCausalLM, VLChatProcessor
 from PIL import Image
 import numpy as np
-import spaces  
+import spaces  # Ensure this is available
 
 # Load the model and processor
 model_path = "deepseek-ai/Janus-Pro-7B"
@@ -17,10 +17,7 @@ vl_gpt = AutoModelForCausalLM.from_pretrained(
     language_config=language_config,
     trust_remote_code=True
 )
-if torch.cuda.is_available():
-    vl_gpt = vl_gpt.to(torch.bfloat16).cuda()
-else:
-    vl_gpt = vl_gpt.to(torch.float16)
+vl_gpt = vl_gpt.to(torch.bfloat16).cuda() if torch.cuda.is_available() else vl_gpt.to(torch.float16)
 
 vl_chat_processor = VLChatProcessor.from_pretrained(model_path)
 tokenizer = vl_chat_processor.tokenizer