AndriusKei commited on
Commit
7fbb3e6
verified
1 Parent(s): 999844b

Update app.py

Browse files

ocr_pipeline = pipeline("image-text-to-text", model="Salesforce/blip2-flan-t5-xl") fix

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- ocr_pipeline = pipeline("image-to-text", model="Salesforce/blip2-flan-t5-xl") # or use 'Salesforce/blip2-opt-2.7b'
5
  classifier = pipeline("text-classification", model="didulantha/sms-spam-detector")
6
 
7
  custom_label_map = {
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ ocr_pipeline = pipeline("image-text-to-text", model="Salesforce/blip2-flan-t5-xl") # or use 'Salesforce/blip2-opt-2.7b'
5
  classifier = pipeline("text-classification", model="didulantha/sms-spam-detector")
6
 
7
  custom_label_map = {