kithangw commited on
Commit
82e19f1
1 Parent(s): 2bf8a61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -5,7 +5,6 @@ import torch
5
 
6
  # Initialize the image-to-text pipeline and models
7
  def load_models():
8
- # Make sure to use the correct model names and tokenizer
9
  image_pipeline = pipeline("image-to-text", model="microsoft/trocr-large-printed")
10
  phishing_model = AutoModelForSequenceClassification.from_pretrained("kithangw/phishing_link_detection1", num_labels=2)
11
  phishing_tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 
5
 
6
  # Initialize the image-to-text pipeline and models
7
  def load_models():
 
8
  image_pipeline = pipeline("image-to-text", model="microsoft/trocr-large-printed")
9
  phishing_model = AutoModelForSequenceClassification.from_pretrained("kithangw/phishing_link_detection1", num_labels=2)
10
  phishing_tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")