Upload app.py

app.py

@@ -13,7 +13,7 @@ README = """
 """
 
 
-model = "roberta-base"
+model_name = "roberta-base"
 learning_rate = 5e-5
 batch_size_train = 64
 step = 1900
@@ -41,9 +41,9 @@ def tokenized_data(tokenizer, inputs):
 
 
 def greet(text):
-    tokenizer = AutoTokenizer.from_pretrained(model)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(
-        model, num_labels=2, id2label=id2label, label2id=label2id,
+        model_name, num_labels=2, id2label=id2label, label2id=label2id,
         state_dict=state_dict
     )
     inputs = tokenized_data(tokenizer, text)
@@ -72,4 +72,4 @@ demo2 = gr.Interface(fn=greet, inputs="text", outputs="text",
                          description=description, examples=examples)
     
 if __name__ == "__main__":
-    demo1.launch()   
+    demo2.launch()