aseifert commited on
Commit
67feba3
1 Parent(s): eadae5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import subprocess
2
  import streamlit as st
3
- from fastT5 import export_and_get_onnx_model, get_onnx_model
4
 
5
  p = subprocess.Popen(["pip", "freeze"], stdout=subprocess.PIPE)
6
  output = p.communicate()[0]
@@ -8,6 +7,8 @@ st.code(output.decode("utf-8"))
8
 
9
  MODEL_NAME = "stas/mt5-tiny-random"
10
 
 
 
11
  model = export_and_get_onnx_model(MODEL_NAME)
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
13
 
1
  import subprocess
2
  import streamlit as st
 
3
 
4
  p = subprocess.Popen(["pip", "freeze"], stdout=subprocess.PIPE)
5
  output = p.communicate()[0]
7
 
8
  MODEL_NAME = "stas/mt5-tiny-random"
9
 
10
+ from fastT5 import export_and_get_onnx_model, get_onnx_model # this line will throw an error
11
+
12
  model = export_and_get_onnx_model(MODEL_NAME)
13
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
14