Uhhy commited on
Commit
de36b22
·
verified ·
1 Parent(s): 50f9f62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ class ModelManager:
52
  def __init__(self):
53
  self.models = []
54
  self.loaded = False
55
-
56
  @spaces.GPU(duration=0)
57
  def load_model(self, model_config):
58
  return {"model": Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename']), "name": model_config['name']}
 
52
  def __init__(self):
53
  self.models = []
54
  self.loaded = False
55
+
56
  @spaces.GPU(duration=0)
57
  def load_model(self, model_config):
58
  return {"model": Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename']), "name": model_config['name']}