Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import time
-#from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import pipeline
 import torch
+#from transformers import AutoModelForCausalLM, AutoTokenizer
 
 #@st.cache(allow_output_mutation=True)
 #def define_model():
@@ -9,8 +10,6 @@ import torch
 #    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b", use_fast=False)
 #    return model, tokenizer
     
-from transformers import pipeline
-
 generator = pipeline('text-generation', model="facebook/opt-1.3b")
 answer = generator("Hello, I'm am conscious and")