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

Update Visualisation/app.py

Browse files
Files changed (1) hide show
  1. Visualisation/app.py +1 -2
Visualisation/app.py CHANGED
@@ -9,8 +9,7 @@ app = FastAPI()
9
 
10
  table_analyzer = pipeline("table-question-answering", model="google/tapas-base")
11
 
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
 
 
9
 
10
  table_analyzer = pipeline("table-question-answering", model="google/tapas-base")
11
 
12
+ user_input_processor = pipeline("text-generation", model="tiiuae/falcon-3b-instruct",device_map="auto",torch_dtype=torch.float16)
 
13
 
14
  image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b",device_map="auto",torch_dtype=torch.float16 )
15