Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from huggingface_hub import InferenceClient
|
|
3 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
4 |
import torch
|
5 |
|
6 |
-
model = GPT2LMHeadModel.from_pretrained('/
|
7 |
tokenizer = GPT2Tokenizer.from_pretrained('openai-community/gpt2-medium')
|
8 |
|
9 |
def chat_with_bot(prompt):
|
|
|
3 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
4 |
import torch
|
5 |
|
6 |
+
model = GPT2LMHeadModel.from_pretrained('raihanp/medical-chatbot')
|
7 |
tokenizer = GPT2Tokenizer.from_pretrained('openai-community/gpt2-medium')
|
8 |
|
9 |
def chat_with_bot(prompt):
|