Hugging Face's logo

Spaces:
tournas
/
storytelling_assistant
Running

App Files Community
tournas commited on
Commit
0ec87e7
·
verified ·
1 Parent(s): 14ac92b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ yolo_model = YOLO("yolov8s.pt")
27
  stable_diffusion = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
28
  stable_diffusion.to(device)
29
  nltk.download("punkt")
30
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
31
 
32
  @spaces.GPU
33
  def detect_objects(image_path):
 
27
  stable_diffusion = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
28
  stable_diffusion.to(device)
29
  nltk.download("punkt")
30
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
31
 
32
  @spaces.GPU
33
  def detect_objects(image_path):