aaliyaan commited on
Commit
82c2eb1
·
1 Parent(s): 33fa42b

Initial commit for JobMatch

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # Load model and tokenizer
5
- tokenizer = AutoTokenizer.from_pretrained("aaliyaan/t5-small-finetuned-career")
6
- model = AutoModelForSeq2SeqLM.from_pretrained("aaliyaan/t5-small-finetuned-career")
7
 
8
  # Function to generate a response from the model and maintain chat history
9
  def chat_with_gpt(user_input, chat_history):
 
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # Load model and tokenizer
5
+ tokenizer = AutoTokenizer.from_pretrained("aaliyaan/t5-small-JobMatch")
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("aaliyaan/t5-small-JobMatch")
7
 
8
  # Function to generate a response from the model and maintain chat history
9
  def chat_with_gpt(user_input, chat_history):
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  huggingface_hub==0.25.2
2
  gradio
3
- transformers
 
 
1
  huggingface_hub==0.25.2
2
  gradio
3
+ transformers
4
+ torch