Update app.py

app.py

@@ -2,13 +2,15 @@
 
 import streamlit as st
 import os
+import torch  # Add this line to import the torch library
+import transformers
+
+
 HF_TOKEN = os.getenv('HF_TOKEN')
 from huggingface_hub import HfFolder
 # Set the token using HfFolder (this persists the token)
 HfFolder.save_token(HF_TOKEN)
 
-import transformers
-
 from transformers import pipeline
 # Load the model, specifying the use of GPU if available
 device = 0 if torch.cuda.is_available() else -1  # use GPU if available