Adding a pytorch import because that's breaking the app in streamlit.com for some reason

test-app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import torch as nn
 import requests
 from transformers import pipeline