Update Visualisation/app.py

Visualisation/app.py

@@ -9,9 +9,9 @@ app = FastAPI()
 
 table_analyzer = pipeline("table-question-answering", model="google/tapas-base")
 
-user_input_processor = pipeline("text-generation", model="tiiuae/falcon-7b-instruct",device_map="auto",torch_dtype=torch.float16)
+user_input_processor = pipeline("text-generation", model="tiiuae/falcon-7b-instruct",torch_dtype=torch.float16)
 
-image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b",device_map="auto",torch_dtype=torch.float16  )
+image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b",torch_dtype=torch.float16  )
 
 # ✅ Load T5 Model (ensure correct architecture)
 model_name = "google/t5-small"  # Change to the correct T5 model if needed