McAwesomeville commited on
Commit
1be65a2
1 Parent(s): 04efaa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,7 +1,10 @@
1
  # app.py
2
 
3
  import streamlit as st
4
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 
 
5
 
6
  def main():
7
  st.title("Hugging Face SQL Generator")
 
1
  # app.py
2
 
3
  import streamlit as st
4
+ from transformers import AutoModelForSeq2SeqLM,
5
+ import torch
6
+ from transformers import pipeline, AutoModelForSeq2SeqLM
7
+
8
 
9
  def main():
10
  st.title("Hugging Face SQL Generator")