ahmed-7124 commited on
Commit
f02b1d8
·
verified ·
1 Parent(s): 84bab0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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)