dammy commited on
Commit
6c36800
·
1 Parent(s): fdf12d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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")