Update app.py

app.py

@@ -19,7 +19,7 @@ import requests
 #from sklearn.metrics.pairwise import cosine_similarity
 
 #device = "cuda:0" if torch.cuda.is_available() else "cpu"
-model_1 = gr.interface.huggingface.load('sentence-transformers/stsb-distilbert-base') 
+model_1 = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
 
 #SentenceTransformer('stsb-distilbert-base', device=device)
 
@@ -46,10 +46,10 @@ def softmax(x):
 
 #model = GPT2LMHeadModel.from_pretrained('distilgpt2', output_hidden_states = True, output_attentions = True)
 
-model  = gr.interface.huggingface.load('distilgpt2', output_hidden_states = True, output_attentions = True)
+model  =  gr.Interface.load('huggingface/distilgpt2', output_hidden_states = True, output_attentions = True)
 
 model.eval()
-tokenizer = gr.interface.huggingface.load('distilgpt2')
+tokenizer =  gr.Interface.load('huggingface/distilgpt2')
 
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 #tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')