codebot

Runtime error

BharadhwajS commited on Jun 25, 2024

Commit

b793553

verified ·

1 Parent(s): b33ac26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 llm = HuggingFaceEndpoint(
     #repo_id="google/gemma-7b-it",
     #repo_id="google/gemma-1.1-7b-it",
-    repo_id="Clinical-AI-Apollo/Medical-NER",
     task="text-generation",
     max_new_tokens=512,
     top_k=5,

 llm = HuggingFaceEndpoint(
     #repo_id="google/gemma-7b-it",
     #repo_id="google/gemma-1.1-7b-it",
+    repo_id="google/gemma-2b",
     task="text-generation",
     max_new_tokens=512,
     top_k=5,