Update app.py

app.py

@@ -1,17 +1,15 @@
+import streamlit as st
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from datasets import load_dataset
+import torch
 
-# Load the BookCorpus74M dataset
+# Load the dataset
 ds = load_dataset("raddwolf/BookCorpus74M")
 
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("FabbriSimo01/Facebook_opt_1.3b_Quantized")
 model = AutoModelForCausalLM.from_pretrained("FabbriSimo01/Facebook_opt_1.3b_Quantized")
 
-import streamlit as st
-import torch
-
 # Streamlit interface
 st.title("Humanized Text Generation App")
 
@@ -34,4 +32,3 @@ if st.button("Generate Humanized Text"):
         st.write(generated_text)
     else:
         st.write("Please enter a valid query.")
-