Spaces:
Sleeping
Sleeping
Update utils.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 = (
|