ikraamkb commited on
Commit
5963e52
Β·
verified Β·
1 Parent(s): 12936e4

Update Visualisation/app.py

Browse files
Files changed (1) hide show
  1. Visualisation/app.py +3 -4
Visualisation/app.py CHANGED
@@ -8,11 +8,10 @@ from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
8
  app = FastAPI()
9
 
10
  # βœ… Load Hugging Face models dynamically from the internet
11
- #code_generator =pipeline("text-generation", model="codellama/CodeLlama-7b-Python")
12
- table_analyzer = pipeline("table-question-answering", model="google/tapas-large")
13
  user_input_processor = pipeline("text-generation", model="tiiuae/falcon-7b-instruct") # comprend lsngusge naturel
14
-
15
- #image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b")
16
 
17
  # βœ… Load T5 Model (ensure correct architecture)
18
  model_name = "google/t5-small" # Change to the correct T5 model if needed
 
8
  app = FastAPI()
9
 
10
  # βœ… Load Hugging Face models dynamically from the internet
11
+ code_generator =pipeline("text-generation", model="codellama/CodeLlama-7b-Python")
12
+ table_analyzer = pipeline("table-question-answering", model="google/tapas-base")
13
  user_input_processor = pipeline("text-generation", model="tiiuae/falcon-7b-instruct") # comprend lsngusge naturel
14
+ image_captioner = pipeline("image-to-text", model="Salesforce/blip2-opt-2.7b")
 
15
 
16
  # βœ… Load T5 Model (ensure correct architecture)
17
  model_name = "google/t5-small" # Change to the correct T5 model if needed