Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ import torch
|
|
9 |
import gradio as gr
|
10 |
import uuid
|
11 |
from sentence_transformers import SentenceTransformer
|
|
|
12 |
|
13 |
# model_name = 'google/flan-t5-base'
|
14 |
# model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|
|
|
9 |
import gradio as gr
|
10 |
import uuid
|
11 |
from sentence_transformers import SentenceTransformer
|
12 |
+
import os
|
13 |
|
14 |
# model_name = 'google/flan-t5-base'
|
15 |
# model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|