shrish191 commited on
Commit
31d12dd
·
verified ·
1 Parent(s): 8524e5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -548,6 +548,8 @@ import numpy as np
548
  import re
549
  import matplotlib.pyplot as plt
550
  import pandas as pd
 
 
551
 
552
  # Load main lightweight model (PyTorch based)
553
  main_model_name = "distilbert-base-uncased-finetuned-sst-2-english"
@@ -561,6 +563,7 @@ fallback_model_name = "cardiffnlp/twitter-roberta-base-sentiment"
561
  fallback_tokenizer = AutoTokenizer.from_pretrained(fallback_model_name)
562
  fallback_model = AutoModelForSequenceClassification.from_pretrained(fallback_model_name).to(device)
563
 
 
564
  # Reddit API setup
565
  reddit = praw.Reddit(
566
  client_id=os.getenv("REDDIT_CLIENT_ID"),
 
548
  import re
549
  import matplotlib.pyplot as plt
550
  import pandas as pd
551
+ # Install tesseract OCR (only runs once in Hugging Face Spaces)
552
+ os.system("apt-get update && apt-get install -y tesseract-ocr")
553
 
554
  # Load main lightweight model (PyTorch based)
555
  main_model_name = "distilbert-base-uncased-finetuned-sst-2-english"
 
563
  fallback_tokenizer = AutoTokenizer.from_pretrained(fallback_model_name)
564
  fallback_model = AutoModelForSequenceClassification.from_pretrained(fallback_model_name).to(device)
565
 
566
+
567
  # Reddit API setup
568
  reddit = praw.Reddit(
569
  client_id=os.getenv("REDDIT_CLIENT_ID"),