Update app.py

app.py

@@ -28,7 +28,7 @@ def build_transform(input_size):
 def preprocess_image(image, input_size=448):
     """Preprocess the image to the required format."""
     transform = build_transform(input_size)
-    tensor_image = transform(image).unsqueeze(0).to(torch.float32).to(device)  # Use float32 for CPU
+    tensor_image = transform(image).unsqueeze(0).to(torch.float32)  # Use float32 for CPU
     return tensor_image
 
 # Load the model and tokenizer
@@ -36,7 +36,6 @@ logging.info("Loading model from Hugging Face Hub...")
 model_path = "OpenGVLab/InternVL2_5-1B"
 model = AutoModel.from_pretrained(
     model_path,
-    torch_dtype=torch.float32,  # Use float32 for CPU compatibility
     trust_remote_code=True,
 ).to(device).eval()
 
@@ -79,4 +78,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    interface.launch(server_name="0.0.0.0", server_port=7860)