aaliyaan commited on
Commit
1ec8383
·
1 Parent(s): c6162d7

Fixed missing import for AutoModelForSeq2SeqLM

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from transformers import AutoTokenizer, AutoModelForCausalLM
3
  from PyPDF2 import PdfReader
4
 
5
  # Models and tokenizers setup
@@ -18,6 +18,7 @@ models = {
18
  },
19
  }
20
 
 
21
  # Function for text generation
22
  def generate_text(model_choice, input_text, max_tokens, temperature, top_p):
23
  model_info = models[model_choice]
 
1
  import gradio as gr
2
+ from transformers import AutoModelForSeq2SeqLM, AutoModelForCausalLM, AutoTokenizer
3
  from PyPDF2 import PdfReader
4
 
5
  # Models and tokenizers setup
 
18
  },
19
  }
20
 
21
+
22
  # Function for text generation
23
  def generate_text(model_choice, input_text, max_tokens, temperature, top_p):
24
  model_info = models[model_choice]