acecalisto3 commited on
Commit
dc59998
·
verified ·
1 Parent(s): e4efcff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -35,7 +35,7 @@ from huggingface_hub import login
35
  from transformers import AutoTokenizer, AutoModelForMaskedLM
36
 
37
  tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2", clean_up_tokenization_spaces=True)
38
- model = AutoModelForMaskedLM.from_pretrained("sentence-transformers/all-mpnet-base-v2")
39
 
40
  # Define classifier for zero-shot classification
41
  classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
@@ -64,7 +64,7 @@ logging.basicConfig(
64
  )
65
 
66
  # Define constants
67
- DEFAULT_FILE_PATH = "scraped_data"
68
  PURPOSE = (
69
  "You go to Culvers sites, you continuously seek changes on them since your last observation. "
70
  "Anything new that gets logged and dumped into csv, stored in your log folder at user/app/scraped_data."
@@ -1246,7 +1246,7 @@ if __name__ == "__main__":
1246
 
1247
  # Create and launch Gradio interface
1248
  demo = create_interface()
1249
- demo.launch()
1250
 
1251
- # Run automated tests
1252
- unittest.main(argv=[''], exit=False)
 
35
  from transformers import AutoTokenizer, AutoModelForMaskedLM
36
 
37
  tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2", clean_up_tokenization_spaces=True)
38
+ model = AutoModelForMaskedLM.from_pretrained("facebook/bart-large-mnli")
39
 
40
  # Define classifier for zero-shot classification
41
  classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
64
  )
65
 
66
  # Define constants
67
+ DEFAULT_FILE_PATH = "user/home/app/scraped_data/<"school name">/"<"school_name_log"/
68
  PURPOSE = (
69
  "You go to Culvers sites, you continuously seek changes on them since your last observation. "
70
  "Anything new that gets logged and dumped into csv, stored in your log folder at user/app/scraped_data."
 
1246
 
1247
  # Create and launch Gradio interface
1248
  demo = create_interface()
1249
+ demo.launch(debug=true)
1250
 
1251
+ # # Run automated tests
1252
+ # unittest.main(argv=[''], exit=False)