Update app.py
Browse files
app.py
CHANGED
|
@@ -181,7 +181,7 @@ class i3HybridModel(nn.Module):
|
|
| 181 |
# ============================================================================
|
| 182 |
|
| 183 |
class SpaceInferenceEngine:
|
| 184 |
-
def __init__(self, repo_id="
|
| 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:
|
| 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 |
)
|