Master0fNone commited on
Commit
446e2f7
·
verified ·
1 Parent(s): 6791153

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import torch
5
  import requests
6
  from PIL import Image
7
  from io import BytesIO
 
8
 
9
  # Import a specific model
10
  model_name = "facebook/bart-large-cnn"
 
5
  import requests
6
  from PIL import Image
7
  from io import BytesIO
8
+ from transformers import pipeline
9
 
10
  # Import a specific model
11
  model_name = "facebook/bart-large-cnn"