Update app.py

app.py

@@ -1,12 +1,15 @@
 import gradio as gr
 import random
+import os
+from huggingface_hub import login
 from transformers import pipeline
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
+login(os.environ["HF_TOKEN"])
 #https://huggingface.co/facebook/opt-1.3b
 #generator = pipeline('text-generation', model="microsoft/DialoGPT-medium")
 tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
 #model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
-model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1',use_auth_token=True)
+model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1')
 
 def create_response(input_str):
   #output_raw= generator(input_str)