imperialwool commited on
Commit
f942112
1 Parent(s): b1b7ebb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = model.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,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 = model.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", " | ") }}
 
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", " | ") }}