Spaces:

VeuReu
/

stools

Running on Zero

VeuReu commited on 9 days ago

Commit

28b1c4e

verified ·

1 Parent(s): f09fbf3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from moe_tools import SalamandraClient
 # ================= Config =================
 MODEL_ID = os.environ.get("MODEL_ID", "BSC-LT/salamandra-7b-tools")

 import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from salamandra_tools import SalamandraClient
 # ================= Config =================
 MODEL_ID = os.environ.get("MODEL_ID", "BSC-LT/salamandra-7b-tools")