Update app.py

app.py

@@ -27,9 +27,13 @@ extra_stopwords = ['ii', 'iii'] # Can add extra stopwords to be removed from dat
 all_stopwords.extend(extra_stopwords)
 
 ########### GET CLAIMED TRAINED MODEL ###########
-auth_token = os.environ.get("AUTH_TOKEN_SECRET")
+auth_token = os.environ.get("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")
 #model_path = os.environ.get("MODEL_PATH")
-model_path = 'eachadea/vicuna-13b'
+model = LlamaForCausalLM.from_pretrained(
+     "Claimed/capybara", use_auth_token=auth_token, 
+    load_in_8bit=True,
+    device_map="auto")
+#model_path = 'eachadea/vicuna-13b'
 tokenizer = LlamaTokenizer.from_pretrained(model_path, use_auth_token=auth_token)
 model = LlamaForCausalLM.from_pretrained(
      model_path, use_auth_token=auth_token, 
@@ -69,7 +73,7 @@ def broad_scope_class_predictor(class_embeddings, abstract_embedding, SearchType
       Links = [f'https://patents.google.com/?q=({x}%2f00)&oq={x}%2f00' for x in HighestSimilarityClass]
     elif SearchType == 'Espacenet Patent Search':
       Links = [f'https://worldwide.espacenet.com/patent/search?q=cpc%3D{x}%2F00%2Flow' for x in HighestSimilarityClass]    
-    HighestSimilarity = pd.DataFrame({'Class':HighestSimilarity, 'Link':Links})
+    HighestSimilarity = pd.DataFrame({'Class':HighestSimilarity})
     return HighestSimilarity