chabane commited on
Commit
c2bbc3a
·
1 Parent(s): 2cb2e14

modify the styles

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -30,7 +30,7 @@ try:
30
  print("[Info] installing Salesforce/blip-image-captioning-base ....")
31
  blip_dir = "./models/blip-base-tf"
32
  snapshot_download("Salesforce/blip-image-captioning-base", local_dir=blip_dir, local_dir_use_symlinks=False)
33
- interpreter = pipeline("image-captioning", model="Salesforce/blip-image-captioning-base")
34
  print("[Info] Salesforce/blip-image-captioning-base is inatalled.")
35
  except Exception as exp:
36
  print("Can't load the model Salesforce/blip-image-captioning-base")
@@ -136,6 +136,8 @@ def summerzation(file:UploadFile=File(...)):
136
  else:
137
  return {"error": "Unsupported file type"}
138
 
 
 
139
  result=""
140
  global summarizer
141
  for i in range(0, len(text), 1024):
 
30
  print("[Info] installing Salesforce/blip-image-captioning-base ....")
31
  blip_dir = "./models/blip-base-tf"
32
  snapshot_download("Salesforce/blip-image-captioning-base", local_dir=blip_dir, local_dir_use_symlinks=False)
33
+ interpreter = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
34
  print("[Info] Salesforce/blip-image-captioning-base is inatalled.")
35
  except Exception as exp:
36
  print("Can't load the model Salesforce/blip-image-captioning-base")
 
136
  else:
137
  return {"error": "Unsupported file type"}
138
 
139
+ print(f'[INFO] text_extracted : {text}' )
140
+
141
  result=""
142
  global summarizer
143
  for i in range(0, len(text), 1024):