Nandoguer commited on
Commit
11f11a1
·
verified ·
1 Parent(s): 3a2c400

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ test_texts = [example['text'] for example in test_data]
41
  test_labels = [example['label'] for example in test_data]
42
  label_mapping = {0: 'sadness', 1: 'joy', 2: 'love', 3: 'anger', 4: 'fear', 5: 'surprise'}
43
  test_labels = [label_mapping[label].lower() for label in test_labels]
44
-
45
  # Evaluate DistilBERT
46
  distilbert_predictions = [distilbert_classifier(text)[0]['label'].lower() for text in test_texts]
47
  print("DistilBERT Model Evaluation Metrics:")
@@ -59,7 +59,7 @@ for text in test_texts:
59
  print("RoBERTa Model Evaluation Metrics:")
60
  print("Accuracy:", accuracy_score(test_labels, roberta_predictions))
61
  print("F1 Score:", f1_score(test_labels, roberta_predictions, average='weighted'))
62
-
63
  # Function to generate an image from a prompt using Stable Diffusion
64
  def generate_image(prompt):
65
  try:
 
41
  test_labels = [example['label'] for example in test_data]
42
  label_mapping = {0: 'sadness', 1: 'joy', 2: 'love', 3: 'anger', 4: 'fear', 5: 'surprise'}
43
  test_labels = [label_mapping[label].lower() for label in test_labels]
44
+ '''
45
  # Evaluate DistilBERT
46
  distilbert_predictions = [distilbert_classifier(text)[0]['label'].lower() for text in test_texts]
47
  print("DistilBERT Model Evaluation Metrics:")
 
59
  print("RoBERTa Model Evaluation Metrics:")
60
  print("Accuracy:", accuracy_score(test_labels, roberta_predictions))
61
  print("F1 Score:", f1_score(test_labels, roberta_predictions, average='weighted'))
62
+ '''
63
  # Function to generate an image from a prompt using Stable Diffusion
64
  def generate_image(prompt):
65
  try: