Spaces:

kithangw
/

testingphishinglink

Sleeping

kithangw commited on Mar 17

Commit

9f55aac

•

1 Parent(s): 4c9cbd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,6 @@ from transformers import pipeline, AutoModelForSequenceClassification, AutoToken
 import torch
 # Initialize the image-to-text pipeline and models
-@st.cache(allow_output_mutation=True)
 def load_models():
     # Make sure to use the correct model names and tokenizer
     image_pipeline = pipeline("image-to-text", model="microsoft/trocr-large-printed")

 import torch
 # Initialize the image-to-text pipeline and models
 def load_models():
     # Make sure to use the correct model names and tokenizer
     image_pipeline = pipeline("image-to-text", model="microsoft/trocr-large-printed")