Spaces:
Runtime error
Runtime error
Update Visualisation/app.py
Browse files- Visualisation/app.py +1 -1
Visualisation/app.py
CHANGED
@@ -12,7 +12,7 @@ table_analyzer = pipeline("table-question-answering", model="google/tapas-base")
|
|
12 |
user_input_processor = pipeline("text-generation", model="tiiuae/falcon-3b-instruct",device_map="auto",torch_dtype=torch.float16 # Use half-precision to save VRAM
|
13 |
)
|
14 |
|
15 |
-
image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b",device_map="auto",torch_dtype=torch.float16
|
16 |
|
17 |
# ✅ Load T5 Model (ensure correct architecture)
|
18 |
model_name = "google/t5-small" # Change to the correct T5 model if needed
|
|
|
12 |
user_input_processor = pipeline("text-generation", model="tiiuae/falcon-3b-instruct",device_map="auto",torch_dtype=torch.float16 # Use half-precision to save VRAM
|
13 |
)
|
14 |
|
15 |
+
image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b",device_map="auto",torch_dtype=torch.float16 )
|
16 |
|
17 |
# ✅ Load T5 Model (ensure correct architecture)
|
18 |
model_name = "google/t5-small" # Change to the correct T5 model if needed
|