Spaces:

ruslanmv
/

Medical-Llama3-Chatbot

Running on Zero

ruslanmv commited on May 15

Commit

4622020

•

1 Parent(s): 9b0ea4a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import torch
-model_name = "ruslanmv/Medical-Llama3-8B""
 device_map = 'auto'
 # Check if GPU is available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import torch
+model_name = "ruslanmv/Medical-Llama3-8B"
 device_map = 'auto'
 # Check if GPU is available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")