ANASAKHTAR commited on
Commit
ab1bf33
·
verified ·
1 Parent(s): fe0c7a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import tempfile
7
 
8
  # Initialize Models
9
  # 1. Text Model
10
- chat_pipeline = pipeline("conversational", model="facebook/blenderbot-400M-distill")
11
 
12
  # 2. Image Model
13
  image_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 
7
 
8
  # Initialize Models
9
  # 1. Text Model
10
+ chat_pipeline = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
11
 
12
  # 2. Image Model
13
  image_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")