kawaa99 commited on
Commit
0abb19d
·
verified ·
1 Parent(s): 085ce89

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -2
utils.py CHANGED
@@ -9,7 +9,6 @@ import streamlit as st
9
  import random
10
  from functools import lru_cache
11
  from transformers import pipeline
12
- import random
13
 
14
  # LOAD IDIOMS
15
  @st.cache_data
@@ -165,7 +164,11 @@ def build_examples_map():
165
  # QUIZ GENERATION
166
 
167
  # Load once
168
- generator = pipeline("text-generation", model="google/flan-t5-base")
 
 
 
 
169
 
170
  def generate_ai_sentence(idiom):
171
  prompt = (
 
9
  import random
10
  from functools import lru_cache
11
  from transformers import pipeline
 
12
 
13
  # LOAD IDIOMS
14
  @st.cache_data
 
164
  # QUIZ GENERATION
165
 
166
  # Load once
167
+ @st.cache_resource
168
+ def load_generator():
169
+ return pipeline("text-generation", model="google/flan-t5-base")
170
+
171
+ generator = load_generator()
172
 
173
  def generate_ai_sentence(idiom):
174
  prompt = (