Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,9 @@ themes=["Pick a Genre", "thriller", "fantasy", "romance", "sci-fi", "mystery", "
|
|
| 5 |
|
| 6 |
def reaction(text, theme):
|
| 7 |
modelName= 'HuggingFaceH4/zephyr-7b-beta'
|
|
|
|
|
|
|
|
|
|
| 8 |
thiscontent=f'Give only one {theme} response to this statement: {text}?'
|
| 9 |
messages=[{'role': 'user', 'content': thiscontent}]
|
| 10 |
client=InferenceClient(model=modelName)
|
|
|
|
| 5 |
|
| 6 |
def reaction(text, theme):
|
| 7 |
modelName= 'HuggingFaceH4/zephyr-7b-beta'
|
| 8 |
+
match theme:
|
| 9 |
+
case "Spanish":
|
| 10 |
+
modelName='DeepESP/gpt2-spanish'
|
| 11 |
thiscontent=f'Give only one {theme} response to this statement: {text}?'
|
| 12 |
messages=[{'role': 'user', 'content': thiscontent}]
|
| 13 |
client=InferenceClient(model=modelName)
|