Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
|
4 |
# Initialize the model
|
5 |
model_name = "sarvamai/sarvam-2b-v0.5"
|
6 |
-
pipe = pipeline("text-generation", model=model_name, device=-1)
|
7 |
|
8 |
# Supported languages
|
9 |
LANGUAGES = ["English", "Bengali", "Gujarati", "Hindi", "Kannada", "Malayalam", "Marathi", "Oriya", "Punjabi", "Tamil", "Telugu"]
|
@@ -40,4 +40,4 @@ iface = gr.ChatInterface(
|
|
40 |
)
|
41 |
|
42 |
# Launch the interface
|
43 |
-
iface.launch()
|
|
|
3 |
|
4 |
# Initialize the model
|
5 |
model_name = "sarvamai/sarvam-2b-v0.5"
|
6 |
+
pipe = pipeline("text-generation", model=model_name, device=-1) # Set device to -1 for CPU
|
7 |
|
8 |
# Supported languages
|
9 |
LANGUAGES = ["English", "Bengali", "Gujarati", "Hindi", "Kannada", "Malayalam", "Marathi", "Oriya", "Punjabi", "Tamil", "Telugu"]
|
|
|
40 |
)
|
41 |
|
42 |
# Launch the interface
|
43 |
+
iface.launch()
|