hipnologo commited on
Commit
d900c5f
1 Parent(s): 5dc0980

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,11 +1,10 @@
1
  import gradio as gr
2
  #gr.Interface.load("models/hipnologo/gpt2-imdb-finetune").launch()
3
-
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
 
6
  def predict_review(text):
7
  # Specify the model name or path
8
- model_name = "models/hipnologo/gpt2-imdb-finetune"
9
 
10
  # Load your model and tokenizer
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
1
  import gradio as gr
2
  #gr.Interface.load("models/hipnologo/gpt2-imdb-finetune").launch()
 
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
  def predict_review(text):
6
  # Specify the model name or path
7
+ model_name = "hipnologo/gpt2-imdb-finetune" # Replace with your model name on the Hugging Face model hub
8
 
9
  # Load your model and tokenizer
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)