Update app.py

app.py

@@ -35,7 +35,7 @@ from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2", clean_up_tokenization_spaces=True)
-model = AutoModelForMaskedLM.from_pretrained("facebook/bart-large-mnli")
+model = AutoModelForMaskedLM.from_pretrained("sentence-transformers/all-mpnet-base-v2")
 
 # Define classifier for zero-shot classification
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
@@ -64,7 +64,7 @@ logging.basicConfig(
 )
 
 # Define constants
-DEFAULT_FILE_PATH = "user/home/app/scraped_data/<"school name">/"<"school_name_log"/
+DEFAULT_FILE_PATH = "scraped_data"
 PURPOSE = (
     "You go to Culvers sites, you continuously seek changes on them since your last observation. "
     "Anything new that gets logged and dumped into csv, stored in your log folder at user/app/scraped_data."
@@ -1180,6 +1180,10 @@ def load_model():
             "text-generation",
             model=model,
             tokenizer=tokenizer,
+            truncation=True,
+            do_sample=True,
+            temperature=0.7,
+            top_p=0.95,
             device=0 if torch.cuda.is_available() else -1,
         )
         logging.info("Model loaded successfully.")
@@ -1234,7 +1238,10 @@ def main():
     model = load_model()
     if model:
         logging.info("Application started successfully.")
+        print("Main function executed")
+        print("Creating interface...")
         demo = create_interface()
+        print("Launching interface...")
         demo.launch(server_name="0.0.0.0", server_port=7860)
     else:
         logging.error("Failed to start the application.")
@@ -1246,7 +1253,7 @@ if __name__ == "__main__":
 
     # Create and launch Gradio interface
     demo = create_interface()
-    demo.launch(debug=true)
+    demo.launch()
 
-  #  # Run automated tests
-  #  unittest.main(argv=[''], exit=False)
+    # Run automated tests
+    unittest.main(argv=[''], exit=False)