Shreyas94 commited on
Commit
e62d662
·
verified ·
1 Parent(s): edd46ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
 
6
  # Initialize Phi-3-vision-128k-instruct tokenizer and model
7
  tokenizer = AutoTokenizer.from_pretrained('microsoft/Phi-3-vision-128k-instruct')
8
- model = AutoModelForSeq2SeqLM.from_pretrained('microsoft/Phi-3-vision-128k-instruct')
9
 
10
  # Function to perform Google search and retrieve URLs, filtering by domain
11
  def google_search(query: str, num_results: int = 10):
 
5
 
6
  # Initialize Phi-3-vision-128k-instruct tokenizer and model
7
  tokenizer = AutoTokenizer.from_pretrained('microsoft/Phi-3-vision-128k-instruct')
8
+ model = AutoModelForSeq2SeqLM.from_pretrained('microsoft/Phi-3-vision-128k-instruct', trust_remote_code=True)
9
 
10
  # Function to perform Google search and retrieve URLs, filtering by domain
11
  def google_search(query: str, num_results: int = 10):