llmahmad commited on
Commit
6eb13db
·
verified ·
1 Parent(s): 174f0a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -4,15 +4,13 @@
4
 
5
  import streamlit as st
6
  from transformers import pipeline
7
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
8
  import torch
9
  import gdown
10
 
11
- import streamlit as st
12
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
13
 
14
  # Load the model and tokenizer
15
- model_path = '.'
16
 
17
  tokenizer = AutoTokenizer.from_pretrained(model_path)
18
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
@@ -43,4 +41,4 @@ if __name__ == "__main__":
43
  </style>
44
  """,
45
  unsafe_allow_html=True
46
- )
 
4
 
5
  import streamlit as st
6
  from transformers import pipeline
 
7
  import torch
8
  import gdown
9
 
 
10
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
11
 
12
  # Load the model and tokenizer
13
+ model_path = '.' # Path to the current directory where files are located
14
 
15
  tokenizer = AutoTokenizer.from_pretrained(model_path)
16
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
41
  </style>
42
  """,
43
  unsafe_allow_html=True
44
+ )