Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -329,8 +329,7 @@ except Exception as e:
|
|
329 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
330 |
|
331 |
def answer_medical_query(query):
|
332 |
-
|
333 |
-
try:
|
334 |
inputs = tokenizer(query, return_tensors="pt")
|
335 |
outputs = model.generate(**inputs, max_new_tokens=150, temperature=0.7, top_k=50)
|
336 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
329 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
330 |
|
331 |
def answer_medical_query(query):
|
332 |
+
try:
|
|
|
333 |
inputs = tokenizer(query, return_tensors="pt")
|
334 |
outputs = model.generate(**inputs, max_new_tokens=150, temperature=0.7, top_k=50)
|
335 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|