Spaces:
Running
Running
Commit
·
f942112
1
Parent(s):
b1b7ebb
Update app.py
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def sentimentAnalys():
|
|
109 |
outputs = sa_m(**inputs)
|
110 |
logits = outputs.logits
|
111 |
predicted_sentiment_index = logits.argmax(dim=1).item()
|
112 |
-
predicted_sentiment =
|
113 |
|
114 |
return {"status": "pass", "predicted_sentiment": predicted_sentiment}
|
115 |
except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }}
|
@@ -125,7 +125,7 @@ def langDetect():
|
|
125 |
outputs = ld_m(**inputs)
|
126 |
logits = outputs.logits
|
127 |
predicted_language_index = logits.argmax(dim=1).item()
|
128 |
-
predicted_language =
|
129 |
|
130 |
return {"status": "pass", "predicted_language": predicted_language}
|
131 |
except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }}
|
|
|
109 |
outputs = sa_m(**inputs)
|
110 |
logits = outputs.logits
|
111 |
predicted_sentiment_index = logits.argmax(dim=1).item()
|
112 |
+
predicted_sentiment = sa_m.config.id2label[predicted_sentiment_index]
|
113 |
|
114 |
return {"status": "pass", "predicted_sentiment": predicted_sentiment}
|
115 |
except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }}
|
|
|
125 |
outputs = ld_m(**inputs)
|
126 |
logits = outputs.logits
|
127 |
predicted_language_index = logits.argmax(dim=1).item()
|
128 |
+
predicted_language = ld_m.config.id2label[predicted_language_index]
|
129 |
|
130 |
return {"status": "pass", "predicted_language": predicted_language}
|
131 |
except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }}
|