FlameF0X commited on
Commit
fd38ccb
·
verified ·
1 Parent(s): fd08ab4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -181,7 +181,7 @@ class i3HybridModel(nn.Module):
181
  # ============================================================================
182
 
183
  class SpaceInferenceEngine:
184
- def __init__(self, repo_id="FlameF0X/i3-200m-v2"):
185
  self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
186
  print(f"Loading model on {self.device}...")
187
 
@@ -403,7 +403,7 @@ with gr.Blocks() as demo:
403
  """
404
  ---
405
  <div style="text-align: center; color: #666;">
406
- <p>Built with ❤️ using Gradio | Model: FlameF0X/i3-200m-v2</p>
407
  </div>
408
  """
409
  )
 
181
  # ============================================================================
182
 
183
  class SpaceInferenceEngine:
184
+ def __init__(self, repo_id="i3-lab/i3-200m-v2"):
185
  self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
186
  print(f"Loading model on {self.device}...")
187
 
 
403
  """
404
  ---
405
  <div style="text-align: center; color: #666;">
406
+ <p>Built with ❤️ from Daniel using Gradio | Model: i3-lab/i3-200m-v2</p>
407
  </div>
408
  """
409
  )