Hudda commited on
Commit
bb47301
·
verified ·
1 Parent(s): 303526c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image
3
  import os
4
  import torch
5
  from transformers import pipeline # Assuming you use a Hugging Face pipeline
6
-
7
  # Define the path to the folder where images are stored
8
  FIRE_SHOT_FOLDER = os.path.join(os.getcwd(), 'FireShot')
9
 
 
3
  import os
4
  import torch
5
  from transformers import pipeline # Assuming you use a Hugging Face pipeline
6
+ pip install torch torchvision torchaudio
7
  # Define the path to the folder where images are stored
8
  FIRE_SHOT_FOLDER = os.path.join(os.getcwd(), 'FireShot')
9