ikraamkb commited on
Commit
463fd6a
·
verified ·
1 Parent(s): db90043

Update Visualisation/app.py

Browse files
Files changed (1) hide show
  1. 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 # Reduce VRAM usage)
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