requ

app.py

@@ -1,7 +1,7 @@
 import random
 import os
 from urllib.parse import urlencode
-from pyngrok import ngrok
+#from pyngrok import ngrok
 
 import streamlit as st
 import streamlit.components.v1 as components
@@ -9,7 +9,7 @@ import torch
 from transformers import pipeline, set_seed
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-import torch
+#import torch
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
 print(
@@ -274,4 +274,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()

requirements.txt

@@ -1,5 +1,6 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
-torch
+torch==2.0.0+cu118
 transformers
 urllib3
-pyngrok
+pyngrok
+torchvision==2.0.0+cu118 torchaudio==2.0.0