Mohinikathro commited on
Commit
3c7f376
·
verified ·
1 Parent(s): 332bb26

changes made to app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import json
2
  import gradio as gr
3
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
4
  import torch
5
  import os
6
  import sys
@@ -10,7 +10,7 @@ import sys
10
  # Device and Model Setup
11
  # ===============================
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
- hf_token = os.environ["HF_TOKEN"]
14
 
15
  #Load the model
16
  model_path = "AI-Mock-Interviewer/T5"
 
1
  import json
2
  import gradio as gr
3
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
  import torch
5
  import os
6
  import sys
 
10
  # Device and Model Setup
11
  # ===============================
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
+ hf_token = os.environ["HF_TOKEN", None]
14
 
15
  #Load the model
16
  model_path = "AI-Mock-Interviewer/T5"