remove torch

app.py

@@ -1,4 +1,4 @@
-import torch
+# import torch
 import pickle
 import streamlit as st
 from transformers import Conversation, pipeline
@@ -19,7 +19,8 @@ if "model_name" not in st.session_state:
     
 
 def get_pipeline(model_name):
-    device = 0 if torch.cuda.is_available() else -1
+    # device = 0 if torch.cuda.is_available() else -1
+    device = -1
     chatbot = pipeline(model=model_name, task="conversational", device=device)
     return chatbot