Muhammadidrees commited on
Commit
a6aca8a
·
verified ·
1 Parent(s): daaddd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -5,15 +5,10 @@ import gradio as gr
5
  from transformers import StoppingCriteria, StoppingCriteriaList
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
8
- from huggingface_hub import login
9
- import os
10
-
11
- # Login using the token stored in repository secrets
12
- login(token=os.getenv("HUGGINGFACE_TOKEN"))
13
  # =============================
14
  # Configuration
15
  # =============================
16
- MODEL_PATH = r"zl111/ChatDoctor"
17
  MAX_NEW_TOKENS = 200
18
  TEMPERATURE = 0.5
19
  TOP_K = 50
 
5
  from transformers import StoppingCriteria, StoppingCriteriaList
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
 
 
 
 
 
8
  # =============================
9
  # Configuration
10
  # =============================
11
+ MODEL_PATH = r"Muhammadidrees/JayConverstionalModel"
12
  MAX_NEW_TOKENS = 200
13
  TEMPERATURE = 0.5
14
  TOP_K = 50