Roberta2024 commited on
Commit
08fe8ae
·
verified ·
1 Parent(s): a794651

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ from ragas.metrics import faithfulness, answer_relevancy, context_relevancy
9
 
10
  # 加載模型
11
  openelm_model = AutoModelForCausalLM.from_pretrained("apple/OpenELM-270M", trust_remote_code=True)
12
- openelm_tokenizer = AutoTokenizer.from_pretrained("apple/OpenELM-270M")
13
 
14
  # Gemini API 設置
15
  GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
@@ -94,4 +94,4 @@ iface = gr.Interface(
94
  )
95
 
96
  if __name__ == "__main__":
97
- iface.launch()
 
9
 
10
  # 加載模型
11
  openelm_model = AutoModelForCausalLM.from_pretrained("apple/OpenELM-270M", trust_remote_code=True)
12
+ openelm_tokenizer = AutoTokenizer.from_pretrained("apple/OpenELM-270M", trust_remote_code=True) # Add trust_remote_code here
13
 
14
  # Gemini API 設置
15
  GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 
94
  )
95
 
96
  if __name__ == "__main__":
97
+ iface.launch()